cpu运行gpu上的pytorch 报错:AssertionError:torch not compiled with cuda enabled——已解决

感觉今天介绍的这种方法可以解决所有这种报错出现的问题

事件发生:

报错:AssertionError:torch not compiled with cuda enabled

cpu运行gpu上的pytorch 报错:AssertionError:torch not compiled with cuda enabled——已解决_第1张图片

解决方法:

后来看到这个代码

parser.add_argument('--test_device', default="cuda:0", type=str,

你可能感兴趣的:(python,java,python,数据库,js,深度学习)