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,则安装成功。