Ubuntu中PyTorch的依赖项有哪些

在Ubuntu中安装PyTorch时，需要确保系统具备一系列依赖项，这些依赖项包括：

系统要求

• 操作系统：Ubuntu 24.04 LTS 或更新版本。
• 处理器：64 位 CPU，至少 2 核（推荐 Intel Core i5 或 AMD Ryzen 5）。
• 内存：最小 4 GB（推荐 8 GB 或更多，适合密集 AI 模型）。
• 存储：至少 10 GB 可用空间（推荐使用 SSD 以提高性能）。
• 显卡：可选，推荐 NVIDIA GPU（至少 4 GB VRAM，用于 TensorFlow 或 PyTorch 的 GPU 加速）。

安装依赖项的命令

sudo apt install -y alsa-utils
sudo apt install -y pciutils
ubuntu-drivers common

安装CUDA Toolkit的命令

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/x86_64/cuda-keyring_1.1-1_all.debs
sudo dpkg -i cuda-keyring_1.1-1_all.debs
sudo apt-get update
sudo apt install cuda-toolkit-12-8

安装cuDNN的命令

从NVIDIA官网下载：https://developer.nvidia.com/cudnn
与CUDA 12.x兼容的cuDNN版本。解压后复制文件到CUDA目录：
sudo cp cudnn-linux-x86_64-8.9.7.29_cuda12-archive/include/cudnn*.h /usr/local/cuda/include/

请注意，具体的依赖项和版本可能会随着PyTorch版本的更新而发生变化。因此，在安装PyTorch之前，建议访问PyTorch的官方网站或GitHub页面，以获取最新的安装指南和依赖项列表。