主要原因是显存不足,可以使用
nvidia-smi命令查看GPU使用情况。详见https://blog.csdn.net/IT_flying625/article/details/103194767
然后寻找显存占用最少的GPU指定使用。
import os os.environ['CUDA_VISIBLE_DEVICES'] = '1'
应该就没问题了,除非没有找到可用的GPU资源,或者模型太大。
本文分享自 作者个人站点/博客 前往查看
如有侵权,请联系 cloudcommunity@tencent.com 删除。
本文参与 腾讯云自媒体同步曝光计划 ,欢迎热爱写作的你一起参与!