基于Anaconda的PyTorch环境的构建

1、创建虚拟环境

#在Anaconda命令窗口运行下列代码

#创建pytorch虚拟环境
conda create -n pytorch python=3.7
	## -n 为参数，后接虚拟环境的名字
	## python=3.7，表示指定python的版本

#激活虚拟环境
conda activate pytorch

#退出虚拟环境
conda deactivate

2、安装PyTorch包（ 使用GPU（CUDA）运行环境的情况 ）

​ ①在windows命令窗口下查看CUDA的版本

# NVIDIA查看CUDA版本
nvidia-smi

​ ②根据CUDA版本在官网选择PyTorch包

#这里选用CUDA 11.3的版本安装PyTorch和相关依赖包
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

	#-c pytorch 表示使用pytorch官网的channel

3、检查是否安装成功

import torch
torch.cuda.is_available()

#结果显示True表示安装成功

4、安装jupyter

由于jupyter默认安装在bash环境下，故需在pytorch环境下安装jupyter

conda install nb_conda
#如果是在bash环境下安装pytorch，则此步骤可省略。 

5、验证jupyter是否安装成功

在命令行窗口切换到pytorch环境并启动jupyter notebook，在新建文件下查看是否有pytorch的环境，并执行第3步的代码进行测试，如果输出结果为True，则安装成功。