问题:RuntimeError: CUDA error: out of memory
解决方法:
执行nvidia-smi,查看显卡运行情况
选择第二列中剩余内存较多的显卡执行程序:
CUDA_VISIBLE_DEVICES=7 python train.py
7为第一列中显卡序号,问题解决
问题:RuntimeError: CUDA error: out of memory
解决方法:
执行nvidia-smi,查看显卡运行情况
选择第二列中剩余内存较多的显卡执行程序:
CUDA_VISIBLE_DEVICES=7 python train.py
7为第一列中显卡序号,问题解决