在conda中安装pytorch

一、创建conda环境

• 打开cmd，使用以下命令可以创建一个名为pytorch的环境，python版本是3.8。

conda create --name=pytorch python=3.8

二、进入环境

• 刚刚创建的环境叫pytorch，所以用以下命令进入这个环境：

activate pytorch

• 如果在命令行行首出现（pytorch）字样，就是成功进入环境了。如下图：
  

• 如果没有出现，那么可以通过anaconda打开这个环境：
  
  

三、安装pytorch

• 在pytorch官网（https://pytorch.org/get-started/previous-versions/）可以查看不同pytorch版本在不同操作系统的各种安装方式。

• 目前pytorch最新版本是1.8.0的。

• 我的显卡的CUDA版本是11，所以这里我使用：

conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge

• 如果显卡不支持CUDA，也可以安装CPU版本的pytorch，例如使用：

conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cpuonly -c pytorch

• 其他版本的pytorch可以去官网了解。

四、测试

• 在这个环境下，在命令行输入：

python

• 然后输入：

import torch
import torchvision
torch.cuda.is_available()

• 如果像下图一样没有报错，就是安装成功了。