如果第一个gpu已经快满了,虽然你用的是有显存剩余的显卡。但是由于会占用第一个显卡的资源,可能还会报显存不够的错误,这个时候,你可以再代码前面加上
CUDA_VISIBLE_DEVICES=5,6 python a.py
上面的56是你自己电脑有空间的gpu。并且你代码里面的0就是指的显卡5,1是指显卡6。字面意思就是当前可见的显卡。
如果第一个gpu已经快满了,虽然你用的是有显存剩余的显卡。但是由于会占用第一个显卡的资源,可能还会报显存不够的错误,这个时候,你可以再代码前面加上
CUDA_VISIBLE_DEVICES=5,6 python a.py
上面的56是你自己电脑有空间的gpu。并且你代码里面的0就是指的显卡5,1是指显卡6。字面意思就是当前可见的显卡。