不要使用python 3.6.0版本
对应这个版本的pytorch包有问题,会出现如下报错:
from torch._C import * ImportError: DLL load failed: 找不到指定的模块
最好换python版本解决
到官网生成安装命令
不要使用野生安装命令,这玩意官方命令选择的版本最适合你的系统:
官方命令加国内源,下载速度快些:
例子:
pip install torch===1.4.0 torchvision===0.5.0 -f https://download.pytorch.org/whl/torch_stable.html -i https://mirrors.ustc.edu.cn/pypi/web/simple/