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 就可以運行！！！

這種方法應該適用於這種報錯：

爲了嘗試這種方法是不是適用於很多情況，我找了也出現這種報錯的一位博主嘗試，果然可以解決。