您当前的位置:网站首页 » 实用技术

栏目导航

站内搜索









如何快捷方便地安装Pytorch

Pytorch是一个用于训练深度学习网络的代码的工具包,然而其安装往往是十分缓慢(而且Bug重重的)。这篇博客中我将与读者分享我在安装Pytorch包时遇到的问题,以及它们的解决方案。

官方提供的方法

几乎所有博客都会让你从官网下载,当然实际中这么下载缓慢而且不稳定(谁试谁知道)。
官方提供方法
在这个界面中选中你电脑的配置,就能得到一个指令,然后在命令行里运行这个代码就能成功安装Pytorch了(然而网络时好时坏是常态)。

所以,如何才能快速便捷地安装Pytorch呢?

我的方法

从它给出的代码中其实我们可以找到一个新的方法

pip install torch==1.7.0+cpu torchvision==0.8.1+cpu torchaudio===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html1

上面是它给定的指令,我们可以看到这句语句中包含了一个网页

这个语句的含义是从这个网页中下载3个.whl文件,然后使用whl文件安装Pytorch包。实际上我们可以手动安装.whl文件,然后再使用.whl文件安装我们最终想要安装的Pytorch包。

网页里的whl文件
上图是网页中的具体内容,可以看到里面有好多.whl文件。这些文件是针对不同电脑环境的不同whl文件,需要按照环境中Python的版本,系统环境来下载。

安装好选定的.whl文件后,可以运行

pip install [下载的whl文件路径+文件名].whl -i http://pypi.mirrors.ustc.edu.cn/simple --trusted-host pypi.mirrors.ustc.edu.cn1

这句语句可以使用刚才在网页中下载的whl文件从清华镜像源下载Pytorch包。

亲测,是有用的,来自于一个下载了无数次Pytorch包,失败了无数次的小白。

最后检测下,成功了!
在这里插入图片描述