如何使用Conda源快速安装PyTorch？

时间：2021-01-28 21:08:51 收藏：0 阅读：321

数据科学交流群，群号：189158789 ，欢迎各位对数据科学感兴趣的小伙伴的加入！

1.切换国内源

1.1.在家目录生成.condarc

conda config --set show_channel_urls yes

1.2.修改.condarc文件

show_channel_urls: true
channels:
  - defaults
channel_alias: https://mirrors.tuna.tsinghua.edu.cn/anaconda
default_channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/simpleitk

1.3.清除索引缓存，保证用的是镜像站提供的索引

conda clean -i

2.安装mamba

conda install mamba

3.安装PyTorch

3.1.到官网获取安装方式：

https://pytorch.org/get-started/locally/

可以看到里面的命令是这个：

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

3.2.使用mamba安装PyTorch

注意这里我们呢要去掉-c，这样才会从默认的清华源来进行安装，并且我们要使用mamba来加速

mamba install pytorch torchvision torchaudio cudatoolkit=10.2

数据科学交流群，群号：189158789 ，欢迎各位对数据科学感兴趣的小伙伴的加入！

原文：https://www.cnblogs.com/purstar/p/14341723.html