RuntimeError: CUDA error: out of memory解决方法

1.如果这个报错后面跟了想要占用多少显存但是不够

看这个

2.直接报错OOM没有说申请多少显存

我的是第二种情况，前一天可以运行的第二天不知道什么情况就报这个错误了
解决方法，加入以下代码：

import os
os.environ["CUDA_VISIBLE_DEVICES"] = '1'

具体原因：使用CUDA_VISIBLE_DEVICES限制使用的GPU。
比如有0,1,2,3号GPU，CUDA_VISIBLE_DEVICES=2,3，则当前进程的可见GPU只有物理上的2、3号GPU，此时它们的编号也对应变成了0、1，即cuda:0对应2号GPU，cuda:1对应3号GPU。
服务器的默认GPU显卡正在使用，需要使用GPU的另一个空闲的显卡；linux里查看GPU的显卡使用状态用代码

这样就可以正常运行了：