今天在服务器上跑代码遇到了这个问题
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:哪里有问题就要去改哪里,根据实际情况修改。