沒有CUDA的Pytorch .backward（）方法

我試圖運行autograd模塊上的Pytorch tutorial中的代碼。然而，當我運行.backwards()電話，我得到的錯誤：沒有CUDA的Pytorch .backward（）方法

cuda runtime error (38) : no CUDA-capable device is detected at torch/csrc/autograd/engine.cpp:359

我固然沒有CUDA功能的設備的時刻成立，但它是我的理解，這不是絕對必要的（至少我沒有在教程的任何地方找到它）。所以我想知道是否有一種方法可以在沒有支持CUDA的GPU的情況下運行代碼。

什麼是你的代碼片段重現錯誤？順便說一句：它應該是'.backward（）'不是'.backwards（）' –

問題是我安裝了一些CUDA驅動程序。卸載後，在新的虛擬環境中重新安裝pytorch，然後重新啓動，一切按預期運行。 –

如果你不使用'.cuda（）'，那麼代碼應該可以在cpu上運行。爲什麼你的代碼導致cuda錯誤？ –

你應該轉移到你使用的CPU的網絡，輸入和標籤：net.cpu(), Variable(inputs.cpu()), Variable(labels.cpu())

2017-05-04 10:34:58 russel

回答