解決pytorch-gpu 安裝失敗方案

導讀	這篇文章主要介紹瞭解決pytorch-gpu 安裝失敗的記錄，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教

ubuntu18 系統上已經配置好gpu加速環境

import torch
x = 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版本下載出錯的問題解決

換了清華的源之後，官網的 是下面的：

conda install pytorch=1.1.0 torchvision=0.3.0 cudatoolkit=9.0 -c pytorch

但是經常會報HTTP error的錯誤，將-c pytorch刪除即可

conda install pytorch=1.1.0 torchvision=0.3.0 cudatoolkit=9.0

原文來自：