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

事件发生:

报错: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 就可以运行!!!

这种方法应该适用于这种报错:

为了尝试这种方法是不是适用于很多情况,我找了也出现这种报错的一位博主尝试,果然可以解决。

Logo

北京人形旗下天工造物具身智能开源社区,聚焦具身天工与慧思开物两大平台

更多推荐