windows conda 安装 PyTorch 切换清华源镜像教程

文章目录

  • 1.进入需要安装PyTorch的环境
  • 2.切换清华源镜像
  • 3.安装对应版本的torch和torchvision
  • 4.验证是否安装成功
  • 1.进入需要安装PyTorch的环境

    在开始菜单找到 Anaconda Prompt 点击进入(在cmd也可)
    默认进入是base环境,首先进入到需要安装 PyTorch 的环境下:
    (这里我的环境名为 scl-lle,修改为自己对应的名称)

    conda activate scl-lle
    

    2.切换清华源镜像

    复制以下代码执行,切换清华源镜像:
    注意:我这里的镜像 url 均为 http ,而非 https ,因为之前试过 https 会报错!

    CondaHTTPError: HTTP 000 CONNECTION FAILED for url
    

    如果遇到了类似的错误,可以尝试将https改为http
    并且是 win-64 环境,在末尾加上 win-64,若是其他系统需修改

    conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64
    conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/win-64
    conda config --set show_channel_urls yes
    conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64
    

    注意:若想恢复默认,可执行:

    conda config --remove-key channels
    

    此时若直接执行命令安装发现安装还是很慢,还需要进入用户目录下,查看 .condarc 文件
    我的目录是:

    C:\Users\mi
    

    以记事本打开 .condarc 文件:

    若最后有 - default 需要删除并保存,否则在安装时会使用默认 channel

    这样镜像就切换成功啦可以进行安装

    3.安装对应版本的torch和torchvision

    https://pytorch.org/get-started/previous-versions/
    找到自己需要安装的torchtorchvision版本号:

    注意命令最后的 -c pytorch 需要删掉。-c pytorch的意思是,安装下载的channel强制为pytorch官网的channel。所以需要删除才能走清华镜像的channel
    即执行:

    conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0
    

    等待安装成功即可:

    4.验证是否安装成功

    最后进行验证是否安装成功:

    python
    import torch
    import torchvision
    

    执行结果如下图所示:

    没有报错即安装成功啦,撒花!

    来源:全幼儿园最聪明

    物联沃分享整理
    物联沃-IOTWORD物联网 » windows conda 安装 PyTorch 切换清华源镜像教程

    发表评论