ubuntu22.04安装cuda/cudnn/pytorch

一、安装独立显卡驱动

略

关于cuda已经不需要单独安装了，只要安装了nvidia驱动，cuda就自动安装完毕。

在终端输入：

nvidia-smi查看驱动是否安装完成及cuda版本

 

 图中可以看到，驱动程序版本为515.65.01，cuda版本为11.7

二、安装cudnn

在终端输入：

sudo apt install nvidia-cudnn

三、进入pytorch官网主页

https://pytorch.org/get-started/locally/

选择合适的版本下载，根据第一步中的信息，应该选择与之最为接近的11.6版

 pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116

复制命令到终端运行即可。

验证：

1.cuda

import torch

print (torch.cuda.is_available())

2.cudnn

import torch

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

print (torch.backends.cudnn.is_available())

如果正常则返回true