ubuntu22.04 RTX4090 cuda cudnn pytorch

Ubuntu 22.04 (Focal Fossa) 操作系统是一个基于Linux的桌面发行版，它以其稳定性、易用性和社区支持而知名。如果你想要在这样的环境下使用RTX 4090显卡，这是一款高端的NVIDIA GPU，专为深度学习和科学计算等高性能任务设计。

CUDA（Compute Unified Device Architecture）是NVIDIA提供的一种并行计算平台和编程模型，用于利用GPU加速各种计算密集型任务，包括机器学习和深度学习。在Ubuntu 22.04上安装CUDA，你需要下载对应版本的CUDA Toolkit，并配置好CUDA环境变量。

CUDNN（Convolutional Neural Networks）是NVIDIA针对深度学习库的一个优化库，特别适用于处理卷积神经网络。安装CUDNN需要先安装CUDA，然后从NVIDIA官网下载并安装相应版本的CUDNN库。

PyTorch是一个开源的深度学习框架，非常适合动态图操作，易于理解和调试。要在Ubuntu 22.04上运行PyTorch，并与CUDA和CUDNN配合，你需要安装Python的torch库，通常会通过pip安装，并确认它链接到了正确的CUDA和CUDNN版本。

安装步骤大致如下：
1. 更新包列表并安装依赖项：

sudo apt-get update
sudo apt-get install -y build-essential cmake git libncurses5-dev pkg-config libopenblas-dev libhdf5-dev libzlib-dev

2. 安装CUDA和CUDNN：
- 下载CUDA Toolkit
- 设置CUDA环境变量
- 安装CUDNN (如果需要的话)

3. 安装Python和PyTorch：

sudo apt-get install python3-pip
pip3 install torch torchvision torchaudio cudatoolkit=11.6 -f https://download.pytorch.org/whl/cu116/torch_stable.html