Ubuntu22.04安装PyTorch1.13.0 GPU版本

目录

一、电脑相关信息

1. 电脑显卡环境：

二、安装Pytorch1.13.0/cu117（GPU版本）

1. 准备：新建虚拟环境

2. 用conda在线安装pytorch1.13.0/cu117（pytorch==1.13.0 torchvision==0.14.0  pytorch-cuda=11.7）：（在pytorch官网的历史版本里找安装命令）

---

机器情况：新机器，自带的有nvidia-smi显卡驱动。

机器：Ubuntu22.04 + 英伟达显卡（GeForce RTX 4090）+ Anaconda3。

目标：安装gpu版本的PyTorch1.13.0/cu117（pytorch==1.13.0 torchvision==0.14.0 pytorch-cuda=11.7）

一、电脑相关信息

1. 电脑显卡环境：

• 显卡：Nvidia GeForce RTX 4090
• 电脑的显卡驱动版本：525.105.17，CUDA Version：12.0

（CUDA Version12.0表示此显卡驱动最高支持cu12.0。）

二、安装Pytorch1.13.0/cu117（GPU版本）

1. 准备：新建虚拟环境

1）首先安装Anaconda。（很简单，自行搜索）

2）其次在用Anaconda建一个虚拟环境。

conda create -n torch1-13 python=3.10
#（torch1-13 表示自己起的环境的名字。）

3）激活虚拟环境。（后续的操作都是在该虚拟环境下进行的。）

conda activate torch1-13

2. 用conda在线安装pytorch1.13.0/cu117（pytorch==1.13.0 torchvision==0.14.0  pytorch-cuda=11.7）：（在pytorch官网的历史版本里找安装命令）

1）在PyTorch官网历史版本链接（ Previous PyTorch Versions | PyTorch）里面，找到安装Pytorch1.13.0/cu117的安装命令，进行在线安装。

安装命令：

# CUDA 11.7
conda install pytorch==1.13.0 torchvision==0.14.0 torchaudio==0.13.0 pytorch-cuda=11.7 -c pytorch -c nvidia

如下图所示：

2）测试：

可以看到torch和torchvision对应的版本：

(base) ubuntu@ubuntu:~$ conda activate torch1-13
(torch1-13) ubuntu@ubuntu:~$ python
Python 3.10.11 (main, Apr 20 2023, 19:02:41) [GCC 11.2.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> import torchvision
>>> torch.__version__
'1.13.0'
>>> torchvision.__version__
'0.14.0'
>>> torch.cuda.is_available()
True
>>> torch.version.cuda
'11.7'
>>> torch.backends.cudnn.version()
8500
>>> exit()
(torch1-13) ubuntu@ubuntu:~$ 

PyTorch1.13.0 GPU版本（pytorch==1.13.0 torchvision==0.14.0 torchaudio==0.13.0 pytorch-cuda=11.7）就安装好了~ 可以愉快使用了~

参考：

Linux安装Pytorch1.8GPU（CUDA11.1）_太阳花的小绿豆的博客-CSDN博客

Pytorch1.10安装记录（CUDA11.3）_太阳花的小绿豆的博客-CSDN博客

更换python源以及anaconda源_anaconda python源_太阳花的小绿豆的博客-CSDN博客