RuntimeError: CUDA error: out of memory

问题:RuntimeError: CUDA error: out of memory

RuntimeError: CUDA error: out of memory_第1张图片解决方法:

执行nvidia-smi,查看显卡运行情况

RuntimeError: CUDA error: out of memory_第2张图片

 选择第二列中剩余内存较多的显卡执行程序:

CUDA_VISIBLE_DEVICES=7 python train.py

7为第一列中显卡序号,问题解决

 

你可能感兴趣的:(训练问题记录,python)