Tips:不建议conda直接下载,速度极慢
!! 推荐将wheel下到本地再进行安装 !!
1 注意版本匹配
链接:https://github.com/pytorch/vision#installation
注意cuda的版本也要对应!!!
2 下载Wheel
https://download.pytorch.org/whl/torch_stable.html
操作流程:
例如我需要安装torch1.7.0
①查表
torch1.7.0对应torch0.8.0或者0.8.1
②下载
进入链接下载上述两个文件,注意cuda版本要相同!!
找到文件后下载安装即可。
3 安装
pip install 路径+文件名