在anaconda使用虚拟环境安装多个版本的pytorch

• 打开Anaconda prompt（anaconda3）
• 创建虚拟环境
• 查看环境
• 激活环境
• 安装所需要的pytorch版本
• 打开jupter notebook
• 删除虚拟环境
• 安装多个版本的pytorch

点击键盘上的win键，找到Anaconda prompt（anaconda3），单击打开

可以看到此时是base环境

在窗口输入conda create -n env_name python=X.X创建python版本为X.X，名字为env_name的虚拟环境。
例如：

conda create -n tensorflow-gpu python=3.8

我当时创建虚拟环境用的是tensorflow-gpu这个名字。

可以输入conda info --envs查看已有的环境。

（我的电脑是win10系统的）

activate tensorflow-gpu

可以看到此时已经进入tensorflow-gpu环境

在官网上找到自己需要的pytorch版本https://pytorch.org/get-started/locally/
将相应的命令复制过去运行即可

pip3 install torch==1.9.1+cu102 torchvision==0.10.1+cu102 torchaudio===0.9.1 -f https://download.pytorch.org/whl/torch_stable.html

输入jupyter notebook命令，回车，这时候打开的notebook就是在虚拟环境运行的了。

jupyter notebook

如果觉得不需要了也可以把虚拟环境删除

remove -n tensorflow-gpu --all

一个环境只能有一个版本的pytorch，需要多个版本的pytorch就创建多个虚拟环境，然后再分别安装pytorch就可以啦。