700字范文,内容丰富有趣,生活中的好帮手!
700字范文 > torch各个版本镜像_Anaconda安装Pytorch镜像详细教程

torch各个版本镜像_Anaconda安装Pytorch镜像详细教程

时间:2024-03-13 11:24:43

相关推荐

torch各个版本镜像_Anaconda安装Pytorch镜像详细教程

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

1.已经安装好Anaconda

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

3.添加清华镜像

>conda config --add channels https://mirrors.tuna./anaconda/pkgs/free/

>conda config --add channels https://mirrors.tuna./anaconda/pkgs/main/

>conda config --set show_channel_urls yes

4.添加Pytorch镜像

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

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

5.安装Pytorch(官网:/)

>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__)

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。