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

pytorch下载地址:https://pytorch.org/
点击下载网址,出现如下画面,点击箭头处可以下载历史版本
软件安装(4)-pytorch1.8.1安装_第1张图片
之前安装了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的最高版本
软件安装(4)-pytorch1.8.1安装_第2张图片

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

软件安装(4)-pytorch1.8.1安装_第3张图片

5.验证安装

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

软件安装(4)-pytorch1.8.1安装_第4张图片

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

conda remove -n pytorch1.8.1 --all

你可能感兴趣的:(深度学习相关软件安装)