在pycharm跑代码时,使用cpu可以跑但是换成gpu就不行

img

img

img


换成gpu debug到父类to(self.device)那一行就执行不下去了,但是cpu就可以继续debug下去,真的非常困惑,不知道从哪里下手解决,希望有大佬帮忙指点下

请说明你的环境,包括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吗

img


我cuda显示是可以使用的