anaconda离线安装pytorch,解决下载过慢的问题

问题描述

我们在安装pytorch的时候，通常是去pytorch官网的旧版本pytorch页面https://pytorch.org/get-started/previous-versions/找到我们所需要的版本，然后复制命令行在我们创建的环境中运行。

conda install pytorch==1.10.0 torchvision==0.11.0 torchaudio==0.10.0 cudatoolkit=11.3 -c pytorch -c conda-forge

命令最后的-c pytorch -c conda-forge指定了从国外的网站进行包的下载，速度很慢导致下载中断，常见的方法都是选择换源，比如清华源进行下载。

添加清华源

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

然后去掉-c pytorch和 -c conda-forge执行命令：

conda install pytorch==1.10.0 torchvision==0.11.0 torchaudio==0.10.0 cudatoolkit=11.3

如果到这一步能够正常安装完，那就再好不过了，但是我的情况是pytorch和cudatoolkit这两个包在下载过程中依然会中断，导致无法安装：

反复尝试多次依旧无法成功。

解决方法

既然已经知道了镜像源安装包的地址，所以可以直接去手动下载我们所需要的包，然后离线安装。

pytorch安装包地址

cudatoolkit安装包地址

然后将下载的这两个包放到anaconda下的pkgs目录下，这个目录存放了所有通过conda下载的包。从环境中进入该目录执行离线安装的命令：

conda install --offline pytorch-1.10.0-py3.8_cuda11.3_cudnn8.2.0_0.tar.bz2
conda install --offline cudatoolkit-11.3.1-h2bc3f7f_2.tar.bz2

最后再执行一下命令：

conda install pytorch==1.10.0 torchvision==0.11.0 torchaudio==0.10.0 cudatoolkit=11.3

确保所有的包都安装完成。conda list查看已安装的包是否有pytorch torchvision torchaudio cudatoolkit ,缺少的包同样可以通过离线下载的方式进行安装。

最后检查以下pytorch是否可用：

python
import torch
torch.__version__
torch.cuda.is_available()

成功！

##########
觉得有用可以点个赞哦！