跑代码时出现这个错误,输入以下命令
$ nvidia-smi -l
原来是卡0内存爆了,于是指定相对空闲的2卡
CUDA_VISIBLE_DEVICES=2 python my_script.py
可以正常运行
解决RuntimeError: CUDA error: out of memory
原文:https://www.cnblogs.com/Tony100K/p/14520567.html