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

港控/mmm° 2023-07-24 03:07 26阅读 0赞

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

## 事件发生： ##

报错：AssertionError:torch not compiled with cuda enabled

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQ1NTA0MTE5_size_16_color_FFFFFF_t_70]

## 解决方法： ##

后来看到这个代码

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

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQ1NTA0MTE5_size_16_color_FFFFFF_t_70 1]  
把 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 就可以运行！！！**

## 这种方法应该适用于这种报错： ##

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

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQ1NTA0MTE5_size_16_color_FFFFFF_t_70]: /images/20230528/0d86dda501c94decb0d9afc7651cb49e.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQ1NTA0MTE5_size_16_color_FFFFFF_t_70 1]: /images/20230528/7c1c680a7a544d99b2e8ba1222ff8c47.png