RuntimeError: CUDA error: out of memory

遇到问题：RuntimeError: CUDA error: out of memory
CUDA kernel errors might be asynchronously reported at some other API call, so the stacktrace below might be incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1.
Compile with `TORCH_USE_CUDA_DSA` to enable device-side assertions.

抓耳挠腮搞了两天，先是看gpu大小，发现0，1，2都占满，3没有，于是使用

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "3"

还是不行。

又以为安装的cuda和torch版本不对应什么的，就又重新搞一遍，发现还是不管用。

又去加各种花里胡哨的指定gpu的代码，还是不行。（崩溃！！！）

点开各种函数界面的代码一个个看，终于在角落里找到罪魁祸首：

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "2"

。。。。。。

之前在另外一个服务器上跑不下心留下的，删掉，成功开跑。