zoukankan      html  css  js  c++  java
  • 如何使用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

  • 相关阅读:
    自学Java0711
    自学Java0710
    自学Java0709
    自学Java0708
    Leetcode刷题集
    网站收集
    674. 最长连续递增序列『简单』
    680. 验证回文字符串 Ⅱ『简单』
    686. 重复叠加字符串匹配『简单』
    693. 交替位二进制数『简单』
  • 原文地址:https://www.cnblogs.com/purstar/p/14341723.html
Copyright © 2011-2022 走看看