剧毒语C 2022-09-15 00:47 采纳率: 33.3%
浏览 325
已结题

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

img

img

img


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

  • 写回答

7条回答 默认 最新

  • 爱晚乏客游 2022-09-15 09:33
    关注

    请说明你的环境，包括torch，torchvision（这两者需要为GPU版本），cuda，cudnn，以及你的显卡型号显存大小，有些情况是安装包冲突或者安装错误，有些情况是你的显存太小，模型加载不进去

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论
查看更多回答(6条)

报告相同问题？

问题事件

  • 系统已结题 9月25日
  • 已采纳回答 9月17日
  • 创建了问题 9月15日