软件安装(4)-pytorch1.8.1安装

pytorch下载地址：https://pytorch.org/
点击下载网址，出现如下画面，点击箭头处可以下载历史版本

之前安装了cuda11.1，这里选择安装cuda11.1的pytorch1.8.1,得到命令行：

conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge

打开cmd,之前下载了anaconda，准备在anaconda里面创建的虚拟环境进行pytorch1.8.1安装。

1.查看安装的虚拟环境：conda info --envs

这里只有anaconda自带的base环境，为了避免破坏base环境，自己创建一个虚拟环境

2.创建自己的虚拟环境：

输入：conda create -n pytorch1.8.1 python=3.8
之后输入：y
-n后面可以自定义，为自己创建的虚拟环境名
python=后面跟的是创建的虚拟环境所用python版本，这里选择3.8，默认下载3.8的最高版本

3.激活与退出自己创建的虚拟环境

激活：

conda activate pytorch1.8.1

退出：

conda deactivate

4.安装gpu版本pytorch1.8.1

！！！安装前一定得激活自己创建的虚拟环境，否则安装在base环境下

conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge

5.验证安装

python
import torch
torch.cuda.is_available()
import torchvision
torch.__version__

6.使用该命令可以删除创建的虚拟环境

conda remove -n pytorch1.8.1 --all