python安装pytorch

百般思念 提交于 2020-03-07 15:48:33

首先贴上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就好了。

 

国内的镜像我没找到,就这样了。

 

标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!