首先贴上pytorch的官网:https://pytorch.org/
不多介绍,直接开始安装步骤:
1、打开https://pytorch.org/get-started/locally/
2、选择对应要安装的版本。
第一行表示稳定版和预览版
第二行表示系统
第三行选择哪种安装方式
第四行语言
第五行CUDA(看你的显卡支不支持)如果没有就选None。
如何查看支持的CUDA的版本,看这篇文章:https://blog.csdn.net/he99774/article/details/104713315
最后一行,打开cmd就可以安装了。
在cmd安装速度奇慢无比,只有几kb每秒,800M的文件要下好几天。
所以我是直接打开https://download.pytorch.org/whl/torch_stable.html
找到对应的版本,打开我的vpn,用浏览器下。
我是下的这两个(自己下要选择对应的python版本和cu还是cpu)
https://download.pytorch.org/whl/cu101/torch-1.4.0-cp35-cp35m-win_amd64.whl
https://download.pytorch.org/whl/cu101/torchvision-0.5.0-cp35-cp35m-win_amd64.whl
然后用pip安装whl就好了。
国内的镜像我没找到,就这样了。
来源:CSDN
作者:小小131
链接:https://blog.csdn.net/he99774/article/details/104713170