jingziguaigaui150_的博客

原创 2021-02-02

问题描述： AttributeError: module ‘torch._C’ has no attribute '_cuda_setDevice 原因： 下载的pytorch版本是cpu版本的而不是gpu版本的 解决办法： 1.安装一个gpu版本的pytorch（理论上可行，但是我用的是第二种方法） 2.针对我代码中的问题，我做了如下操作： 2.1.`注释掉.cuda(0)改为如下代码： model.to(torch.device('cpu')) #model.cuda(0) 2.2将