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

Windows下Anaconda中安装pytorch_第1张图片

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

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

pip list

如下图,可以看见,已经安装成功pytorch1.7.1
Windows下Anaconda中安装pytorch_第2张图片
六、导入使用

python

进入python环境后,使用

import torch

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

torch.cuda.is_available()

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

你可能感兴趣的:(人工智能,Pytorch)