Ubuntu22.04.3安装PyTorch12.1

一、安装pytorch

1.创建虚拟环境

终端输入如下一行代码

conda create -n torch12.1cu12.2

2.激活并进入虚拟环境

终端输入如下一行代码

conda activate torch12.1cu12.2

3.安装pytorch

终端输入如下一行代码

conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch-c nvidia

二、检验

1.在终端打开python

终端输入如下一行代码

python

2.调用torch进行检验

终端输入如下四行代码

import torch

print(torch.__version__)

print(torch.version.cuda)

print(torch.backends.cudnn.version())

3.输出

如图所示即可