win10安装pytorch简易教程


由于最近重装了电脑系统,环境需要重新配置,所以来这里做个笔记。

1、安装Anaconda

百度搜索清华镜像源,进入下载Anaconda安装文件,直接看我下载的,
anaconda
点击exe文件一路next到底即可,中间注意勾选添加环境变量。
然后cmd进入命令行,输入:

conda create -n pytorch python=3.6.10

会在Anaconda安装目录下的envs中生成一个文件夹,其名为pytorch,其中的python环境是python 3.6.10,这个你可以随便取,只要python官网有就行。
然后在安装之前可以切换conda镜像源

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

然后删除默认镜像源,必须删了!!!

conda config --remove channels defaults 

2、安装pytorch

首先进入刚刚创建好的虚拟环境,

conda activate pytorch

然后点击进入pytorch官网,选择相应版本,这里我选择的是windows、cuda10.2、python版本:
配置图片
直接复制其中的命令到终端即可进行安装。

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch
conda install pytorch torchvision cudatoolkit=10.2

注意,务必去除-c pytorch这串字符,要不然还是用的pytorch官网的镜像源,换源就毫无用处。

接下来静静等待安装完成。
等待片刻,如果安装成功,那么进入pytorch虚拟环境,输入python,回车。
然后输入

import torch # 不报错
import torchvision # 不报错
torch.cuda.is_available() # 返回值应该是True

成功截图

3、小技巧

如果安装失败,没关系,深吸口气,博主这里很贴心,只需要完成以下几步,一般可以成功装上。
第一步:
删除所创建的虚拟环境

conda remove -n pytorch --all

第二步:
卸载anaconda
第三步:
重启电脑
第四步:
打开本博客,重新来一遍。

预祝大家都能装成功!

评论 10
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值