在安装Pytorch时,若使用官网下载路径十分缓慢,一般情况下大家都会使用国内镜像进行下载。使用镜像源方式安装的人比较多,但是也存在安装不了的情况,比如镜像源中没有相关的安装包。
如下,使用Pytorch官网下载速度十分慢,要三个多小时。
而想要切换为其他国内镜像时,找不到需要安装的文件。
这里我安装Pytorch使用了另外一种方案,就是下载它的*.whl文件。Pytorch相关资源链接如下:download.pytorch.org/whl/torch/。进入链接后寻找自己需要的Pytorch文件。CPU版本的Cuda在torch版本号的“+”后面显示“cpu”,若是GPU版本的Cuda则在“+”后面显示“cu”。
直接点击要下载的文件会发现,下载速度与使用命令行速度基本一致。
为了加速,我在这里使用了下载工具IDM,全称是Internet Download Manager,是一种多线程下载工具,可以提升下载文件的速度。IDM链接如下:https://pan.baidu.com/s/1F2M9DHJsboK4eFc-h7HWQw?pwd=hn5b 注意:仅供学习,请勿使用于不合规的用途,否则后果自负。
下载完成后双击安装程序即可,安装完成后,在打开浏览器时会自动添加浏览器扩展,若没有根据说明书添加。
将鼠标移动到需要下载的资源上面,右键选择使用IDM下载或者是复制链接到IDM下载。
使用此方法下载速度能达到4mb/s,相较于原来的263kb/s提升了16倍(16个线程) 。
下载完成后,通过如下命令行安装即可:
pip install 文件路径+文件名
conda install 文件路径+文件名
二选一即可实现安装。
但是,这个方法只安装了torch,并没有安装torchvision和torchaudio,由于我们将最大的torch安装了,这两部分可以直接通过pip install 的方法安装即可。
注意:在安装时一定要指定版本,否则会出现版本不匹配的报错。
比如,我需要安装的pytorch在官网获得的命令如下:
pip install torch==1.10.1+cu111 torchvision==0.11.2+cu111 torchaudio==0.10.1 -f https://download.pytorch.org/whl/cu111/torch_stable.html
那么我就需要在需要安装的环境中或者是在conda中通过命令安装。
pip install torchvision==0.11.2
pip install torchaudio==0.10.1