torch-torchvision-python快速安装

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 路径+文件名

猜你喜欢

转载自blog.csdn.net/lafsca5/article/details/129698199