pytorch 和pytorchvision 安装(conda 清华源 CPU)

1.conda 换源(清华源)

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes

注意:添加完后,找到 .condarc 文件,删除里面的 defaults,这样能快点。(.condarc 为隐藏文件,一般在Home文件夹,Ctrl+H显示隐藏文件)

当然也可以设置清华源优先。(未亲测)

# 优先使用清华conda镜像
conda config --prepend channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ 

2.安装pytorch 和pytorchvision

pytorch官网 ,选择自己的配置(我的配置如图)

生成命令: 

conda install pytorch-cpu torchvision-cpu -c pytorch

注意,此时若直接将命令复制至终端运行,用的依然是默认的https://conda.anaconda.org/pytorch/...URL,因为conda install-c参数表示指定对应的channel安装,速度非常慢。

我们单位竟然把这个URL屏蔽了,根本下载不了!!!否则我也不会折腾换源了!!!

Downloading and Extracting Packages
pytorch-cpu-1.1.0    | 53.6 MB   |                                       |   0%
torchvision-cpu-0.3. | 3.8 MB    |                                       |   0%

CondaHTTPError: HTTP 000 CONNECTION FAILED for url <https://conda.anaconda.org/pytorch/linux-64/pytorch-cpu-1.1.0-py3.7_cpu_0.tar.bz2>
Elapsed: -

An HTTP error occurred when trying to retrieve this URL.
HTTP errors are often intermittent, and a simple retry will get you on your way.

CondaHTTPError: HTTP 000 CONNECTION FAILED for url <https://conda.anaconda.org/pytorch/linux-64/torchvision-cpu-0.3.0-py37_cuNone_1.tar.bz2>

 

删除安装命令最后的 -c pytorch,才会采用清华源安装。

conda install pytorch-cpu torchvision-cpu

可以看到源地址已经切换。

The following packages will be downloaded:

    package                    |            build
    ---------------------------|-----------------
    pytorch-cpu-1.1.0          |      py3.7_cpu_0        53.6 MB  https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud//pytorch
    torchvision-cpu-0.3.0      |    py37_cuNone_1         3.8 MB  https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud//pytorch
    ------------------------------------------------------------
                                           Total:        57.4 MB

安装完成后在终端输入python,然后

>>> import torch
>>> import torchvision

不报错表示成功。

 

 

  • 18
    点赞
  • 47
    收藏
    觉得还不错? 一键收藏
  • 6
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 6
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值