请说明你的环境,包括torch,torchvision(这两者需要为GPU版本),cuda,cudnn,以及你的显卡型号显存大小,有些情况是安装包冲突或者安装错误,有些情况是你的显存太小,模型加载不进去
不debug时,你的GPU可以使用吗?
是不是GPU调用失败走不下去
GPU查看能不能调用:
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu' )
请检查你的cuda,cudnn等资源是否都下载了,具体请参考以下文章http://t.csdn.cn/NSecB
你的电脑支持GPU吗