温馨提示×

Ubuntu中PyTorch的依赖项有哪些

小樊
70
2025-05-18 06:30:42
栏目: 智能运维

在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页面,以获取最新的安装指南和依赖项列表。

0