logo
publist
写文章

简介

该用户还未填写简介

擅长的技术栈

可提供的服务

暂无可提供的服务

【新手】复现NeRCo代码中出现的torch.cuda.OutOfMemoryError: CUDA out of memory. 问题解决办法。

3、高版本pytorch可能在处理显存占用时有更多的优化,可以升级为更高版本的pytorch。特别是在相同型号GPU的情况下,可以考虑Python包版本的问题,通过“pip list”或者“conda list”可以参考版本信息。对于1,在尝试减少batchsize解决,发现原文中原本的batchsize已经是1,没有办法再减了。但是此方法只是应急用,至于怎么让他多个GPU一起用,我至今还没有弄明

#python#图像处理#目标检测 +1
到底了