環境
我是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
完美安裝!!!