這幾天爲了運行python的圖像轉換的項目,不得不安裝pytorch,安裝了兩天,最後把經驗記錄一下。python
若是版本不匹配會拋出不少錯誤,而網上的各類解決方式有大部分也解決不了問題。spa
在安裝pytorch以前,首先是你的PC安裝的是NVIDIA顯卡,而後打開NVIDIA控制面板,查看CUDA的版本,老的顯卡建議就到8.0,由於9.0的cuda要求顯卡的計算能力在3.5以上.net
個人顯卡計算能力只有3.0。若是顯卡的版本較低須要進行cuda版本升級。3d
運算能力參照:https://blog.csdn.net/real_myth/article/details/44308169blog
NVIDIA版本下載:https://developer.nvidia.com/cuda-toolkit-archiveit
顯卡版本安裝以後,須要安裝Anaconda3,參照網上的例子若是這時候進行pytorch安裝,會默認安裝cuda9.1,即便你指定cuda80也很差用,因此只能下載離線版本的class
cuda爲80版本的pytorch。終端
https://pan.baidu.com/s/1dF6ayLr?errno=0&errmsg=Auth%20Login%20Sucess&&bduss=&ssnerror=0&traceid=#list/path=%2Fpytorch下載
下載到本地以後,在Anaconda Prompt終端運行:numpy
conda install numpy mkl cffi
conda install --offline pytorch-0.2.1-py36he6bf560_0.2.1cu80.tar.bz2 #離線安裝
完了!