问题描述
使用pytorch运行Unet代码报错,RuntimeError: CUDA error: out of memory,
可能由于GPU内存被占满而造成的,通过重启Pycharm如未解决问题,可采取如下方案。
解决方案:
将num_workers调小点,如果不行就调到0,实测解决问题。
之前遇到过相似问题,在跑Googlenet神经网络时也遇到报错:[WinError 1455] 页面文件太小,无法完成操作。 Error loading。
首先通过减小batch_size的大小,解决问题,如果问题未得到解决,实测也可以通过将num_workers设为0.问题得到解决。
例如设置成:
//
num_workers=0;