cpu运行gpu上的pytorch 报错:AssertionError:torch not compiled with cuda enabled——已解决
·
感觉今天介绍的这种方法可以解决所有这种报错出现的问题
事件发生:
报错:AssertionError:torch not compiled with cuda enabled

解决方法:
后来看到这个代码
parser.add_argument('--test_device', default="cuda:0", type=str,
help='cuda:0 or cpu')

把 cuda 换成 cpu
parser.add_argument('--test_device', default="cpu:0", type=str,
help='cuda:0 or cpu')
%上面的 default="cuda:0" 改成 default="cpu:0",
下面的 help='cuda:0 or cpu' 因为定义了可以使用 cpu 所以就不用改了
当然也可以改成:
parser.add_argument('--test_device', default="cpu:0", type=str,
help='cpu:0')
parser.add_argument('--test_device', default="cuda:0 or cpu", type=str,
help='cuda:0 or cpu')
总结:
总结一下就是:出现的 cuda 全部换成 cpu 就可以运行!!!
这种方法应该适用于这种报错:
为了尝试这种方法是不是适用于很多情况,我找了也出现这种报错的一位博主尝试,果然可以解决。
更多推荐
所有评论(0)