可能1是显卡不兼容。可能2是显存碎片问题。你可以试试把一整块4G大小的tensor放到gpu上看看会不会报错。
https://discuss.pytorch.org/t/gpu-is-not-utilized-while-occur-runtimeerror-cuda-runtime-error-out-of-memory-at/34780 因此最好的方法就是运行的时候使用CUDA_VISIBLE_DEVICES限制一下使用的GPU。 比如有0,1,2,3号GPU,CUDA_VISIBLE_DEVICES=2,3,则当前进程的可见GPU只有物理上的2、3号GPU,此时它们的...
先说一下想要跑的是什么模型，一运行就错也可能只是单纯的模型无法完全加载到显存
pytorch-CycleGAN-and-pix2pix 强行在win10环境中启动报错 RuntimeError: CUDA out of memory , 需要分配几M内存,还剩几个G内存无法分配。。。 处理了上面的warning OSError: [WinError 1455] 页面文件太小,无法完成操作。 处理 1.重启pycharm(基本没啥用) 2.把num_works设置为0 3.调大页面文件的大小 【这...
可能1是显卡不兼容。可能2是显存碎片问题。你可以试试把一整块4G大小的tensor放到gpu上看看会不会报错。
信息不完全。用的什么模型和optimizer？