问题: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为第一列中显卡序号,问题解决
6万+
3830

被折叠的 条评论
为什么被折叠?