這篇文章主要介紹了解決pytorch-gpu 安裝失敗的記錄,具有很好的參考價值,希望對大家有所幫助。如有錯誤或未考慮完全的地方,望不吝賜教
ubuntu18 系統上已經配置好gpu加速環境,
1 2 3 | import torchx = torch.Tensor(2,3)x_cuda = x.cuda() |
x可以正常,當定義x_cuda 后,命令行卡住,不再繼續輸出
解決方法正在尋找中。。。找到后將記錄解決方案。
當初pytorch-gpu為1.0版本,然而配置的cuda和cudnn版本較低,不支持高版本的pytorch,后來選擇安裝 0.4.1版本的pytorch,解決此問題。
當前環境是 torch-0.4.1-cp36 +384顯卡驅動+cuda8.0+cudnn6.0。
幾多艱辛,得以開顏。
補充:conda安裝pytorch-gpu版本下載出錯的問題解決
換了清華的源之后,官網的命令是下面的:
1 | conda install pytorch=1.1.0 torchvision=0.3.0 cudatoolkit=9.0 -c pytorch |
但是經常會報HTTP error的錯誤,將-c pytorch刪除即可
1 | conda install pytorch=1.1.0 torchvision=0.3.0 cudatoolkit=9.0 |
以上為個人經驗,希望能給大家一個參考