RuntimeError:Cuda error:out of memory解决办法

今天在服务器上跑代码遇到了这个问题

RuntimeError:Cuda error:out of memory

用nvidia-smi看，发现第一块内存不足，是有人在跑代码了，为了选用第二块，于是修改了两个地方：
predict.sh文件中修改CUDA_VISIBLE_DEVICES

CUDA_VISIBLE_DEVICES=1

然后进入predict.py文件中修改语句：

os.environ['CUDA_VISIBLE_DEVICES'] = str('1,0')

这样就可以啦（记录一下今天解决的小问题）
PS：哪里有问题就要去改哪里，根据实际情况修改。