Windows下Anaconda中安装pytorch

前提

在Anaconda Prompt中建立新环境(python3.6版本的)建立新环境的目的,是为了适应不同版本的切换,比如,你想使用3.6版本的python,就切换到3.6的。使用3.7的就切换到3.7的。

conda create -n 当前环境名称 python=3.6 

安装好后,会有提示:
To activate this environment, use(激活环境)

conda activate 环境名称

To deactivate an active environment, use(退出环境)

conda deactivate

安装pytorch

一、首先打开Anaconda Prompt

二、创建一个python3.6的虚拟环境

conda create -n pytorch python=3.6

三、进入虚拟环境

conda activate pytorch

四、虚拟环境下执行(我使用的是cpu版本的)

conda install pytorch torchvision torchaudio cpuonly -c pytorch

在这里插入图片描述

这个命令是到pytorch官网中,根据你想要得gpu或者cpu版本去复制。一般,推荐liunx使用pip,windows使用conda。

如果选择gpu版本的话,要先看看自己的显卡型号,然后选择对应的cuda。推荐看一下B站视频
五、检查是否安装成功

pip list

如下图,可以看见,已经安装成功pytorch1.7.1
在这里插入图片描述
六、导入使用

python

进入python环境后,使用

import torch

导入torch,如果不报错,说明安装成功。
然后在执行,判断是否可用gpu

torch.cuda.is_available()

显示False是不可用(因为我安装是cpu版本的),True是可用。
在这里插入图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值