前提
在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是可用。