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

数据科学交流群,群号: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作为使用最为广泛的数据科学环境管理工具,可以协助我们很方便的完成创建管理环境、下载安装第三方库、软件包等操作,但其在下载资源的过程中下载速度时常令人捉急,即使使用连接速度更快的国内镜像,也摆脱不了其单线程挨个下载资源导致的低效问题。

Mamba(黑曼巴)专为加速Conda而生,其改写了Conda下载资源的固有方式,以多线程的方式对网络资源进行并行下载,从而大幅提升Conda效率,以下是安装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

4.离线安装PyTorch

如果你觉得mamba的安装还是比较慢,可以使用专业的下载工具,比如迅雷。

但是在下载之前你要明白你要知道安装包的下载地址:

 如上图所示,包在解析的时候一般会告诉你地址,用浏览器直接访问这些地址,使用Ctrl+F来搜索具体的包名即可找到。

下载好了以后我们可以使用

conda install --use-local ……

把……替换成你的文件的绝对路径名,如果是windows10,你可以直接按住shift+鼠标右键,然后点击“复制为路径”即可

此时粘贴到命令行终端即可。例如:

conda install --user-local "F:迅雷云盘pytorch-1.7.1-py3.8_cuda102_cudnn7_0.tar.bz2"

注意:由于反复多次下载,很有可能会产生上次的下载内容会对下一次安装产生问题,此时,可以看看报错信息,找到对应的路径删除同名的文件夹即可,然后再用conda install --use-local 来安装就不会再报错了!!!

5.验证一下安装:

进入python环境中:

python

运行:

import torch
x = torch.rand(5, 3)
print(x)

检查Cuda:

import torch
torch.cuda.is_available()

最后,祝你快快安装成功,以便能够进入深度学习的学习阶段!!!

数据科学交流群,群号:189158789 ,欢迎各位对数据科学感兴趣的小伙伴的加入!

深信积累的力量,时间就是你最好的朋友,否则它就是你最大的敌人。

如果你想分享此文章,请注明:作者:PurStar 出处:www.cnblogs.com/purstar

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