前言

安装Pytorch，首先需要安装的是torch，做CV的常用的还有TorchVision，还有必不可少的cudatoolkit。

它们之间的版本有着严格的对应关系。

而官网提供的命令没有涵盖我们所需要的一些版本的对应（如下图所示），所以我们只能模仿官网的命令来下载。

我现在使用的是CUDA11.2，那么我要么使用cudatoolkit11.1和cudatoolkit11.2，而不能使用cudatoolkit11.3，因为不能向上兼容，只能向下兼容，升级CUDA很麻烦，我不想这么麻烦。

安装torch和torchvision

torch和torchvision对应关系

pip install torch==1.9.1+cu111 torchvision==0.10.1+cu111 -f https://download.pytorch.org/whl/torch_stable.html

注：conda install 可能找不到源包，那就在虚拟环境中用 pip install.

安装cudatoolkit

配置清华镜像：

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
anaconda show Zimmf/cudatoolkit

查找cudatoolkit：

anaconda search -t conda cudatoolkit

找到我需要的11.1版本的那一行：

anaconda show Zimmf/cudatoolkit

发现它里面有10.2和11.1的版本，选择我需要的11.1，安装：

conda install --channel https://conda.anaconda.org/Zimmf cudatoolkit=11.1