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