這篇文章主要介紹了解決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 |
以上為個人經驗，希望能給大家一個參考