Win10下快速下载Pytorch(国内镜像)

创建一个单独的环境,用于运行pytorch

conda create -n pytorch python=3.7 #-n pytorch的意思是创建一个名字叫pytorch的环境 

在这里插入图片描述
进入pytorch的官网https://pytorch.org/

在这里插入图片描述
找到适合自己的cuda环境以及所需要的版本

在这里插入图片描述
在命令行下进行如下安装
在这里插入图片描述
但是总是安装不成功,作者一直在qiang外,但是还是没有成功安装,于是使用了清华镜像源进行安装。

通过百度,搜索总结如下利用国内镜像的方式:

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

然后输入如下:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
然后输入如下命令,备注,当我们选择cudatoolkit不同的版本时候,就选择不同的pytorch版本,作者cuda是10.0版本因此选择了cudatoolkit=10.0 默认现在了清华镜像目前收集的最新的pytorch版本,版本是pytorch1.2(截至目前2020.2.27,pytorch最新的版本是1.4版本)

conda install pytorch torchvision cudatoolkit=10.0 -c python

当然也可以自己指定版本,如下所示:

conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=10.0 -c pytorch

另外,当我们完成下载后,需要进行下载成功验证

import torch
print(torch.cuda.is_available()

返回结果为True,则pytorch的GPU版本安装成功

  • 6
    点赞
  • 21
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值