环境
我是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
完美安装!!!