安裝PyTorch最簡單步驟(防踩坑)

環境

我是Windows 7; CUDA 9.0; python 3.6.7;

 

安裝

理論上只要執行以下兩條命令就可以:

pip3 install https://download.pytorch.org/whl/cu90/torch-1.1.0-cp36-cp36m-win_amd64.whl
pip3 install https://download.pytorch.org/whl/cu90/torchvision-0.3.0-cp36-cp36m-win_amd64.whl

爲防止直接使用命令直接安裝由於網絡原因導致下載超時或者中斷而安裝失敗,先離線下載安裝文件:

(1)https://download.pytorch.org/whl/cu90/torch-1.1.0-cp36-cp36m-win_amd64.whl
(2)https://download.pytorch.org/whl/cu90/torchvision-0.3.0-cp36-cp36m-win_amd64.whl

再在文件夾中安裝。

 

問題

一般安裝完運行會出現:ModuleNotFoundError: No module named 'numpy.core._multiarray_umath' 錯誤

這是因爲電腦裏的numpy跟不上時代步伐,要更新了,一般通過鏡像更新:

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple --upgrade numpy

完美安裝!!!

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章