Anaconda安装Pytorch镜像详细教程

冬风吹战鼓擂IP属地: 河北
0.143字数 457阅读 28,077

在win10基础上已经安装好了Anaconda,继续安装Pytorch。因为采用原本的下载方法实在太慢,还容易出现问题,现采取Pytorch清华镜像下载方法,亲测有效快速!!!

1.已经安装好Anaconda

2.搜索cmd>>>打开Anaconda Prompt

3.添加清华镜像

>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

4.添加Pytorch镜像

>conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/

注意:这里有个坑,必须是进到win-64文件夹才可以!!!

5.安装Pytorch(官网:https://pytorch.org/)

>conda install pytorch torchvision cudatoolkit=10.0

注意:博主的cuda是10.0版本,根据不同的版本搜索自己的下载语句,记住去掉-c pytorch

-c pytorch参数指定了conda获取pytorch的channel,在此指定为conda自带的pytorch仓库。我们安装的时候去掉-c pytorch就可以使用清华镜像快速安装。

附:cuda版本查询

>>>打开控制面板>>>点击查看方式>>>选择小图标

>>>点击NVIDIA控制面板

>>>点击左下角系统信息

>>>选择组件>>>查看cuda版本

6.安装完成

7.检测是否安装成功

>>>打开cmd>>>进入python

输入:>>>import torch 

输入:>>>a = torch.Tensor([1.]) 

输入:>>>a.cuda() 

#正常则返回"tensor([ 1.], device='cuda:0')"

输入:>>>from torch.backends import cudnn

输入:>>>cudnn.is_acceptable(a.cuda())   

 #正常则返回 "True"

另:查询pytorch版本

>>>print(torch.__version__)

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。
11人点赞
更多精彩内容,就在简书APP
"小礼物走一走,来简书关注我"
还没有人赞赏,支持一下
总资产0.286共写了1055字获得14个赞共8个粉丝

推荐阅读更多精彩内容