主要原因是显存不足,可以使用
nvidia-smi命令查看GPU使用情况。详见https://blog.csdn.net/IT_flying625/article/details/103194767
然后寻找显存占用最少的GPU指定使用。
import os os.environ['CUDA_VISIBLE_DEVICES'] = '1'
应该就没问题了,除非没有找到可用的GPU资源,或者模型太大。