- 打开https://pytorch.org/
- 查看安装命令:conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch-lts -c nvidia
- 最后的-c代表的是channels 从pytorch官网下载,导致速度慢 所以我们要从国内镜像源中寻找我们需要的pytorch包 例如清华源 https://mirrors.tuna.tsinghua.edu.cn/help/anaconda/
- 查看完整列表我们可以看到有很多包 最后查询到pytorch包在下面的路径下 https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
- 使用命令将上述网址添加conda源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
- 最后使用新的安装命令,将-c pytorch去掉,表示从我们配置的新路径中下载
conda install pytorch torchvision torchaudio cudatoolkit=11.1
|