報錯:AssertionError: Torch not compiled with CUDA enabled

環境:win0+torch(cpu) 詳細報錯信息: 解決: 查閱資料發現此錯誤是由於下載的torch沒有cuda,所以在運行時會報錯,在程序最開始的地方加上: device = torch.device(「cuda」 if torch.cuda.is_available() else 「cpu」) 代碼其餘地方出現.cuda()的地方改成.to(device)就可以在無gpu的環境中運行了。 參
相關文章
相關標籤/搜索