nvidia@ubuntu基于apt安装cuda-driver/cuda-toolkit

Posted on Views:

在 Ubuntu 22.04.3 上通过 apt 安装 NVIDIA A100 GPU 的开发支持库（如 CUDA、cuDNN、NCCL 等），推荐使用 NVIDIA 官方的 APT 仓库来安装。

env

  • ubuntu22.04.3 LTS

1. 添加 NVIDIA APT 仓库（推荐官方源）

1
2
3
4
5
6
7
8
sudo apt update
sudo apt install -y wget gnupg lsb-release

distribution=$(. /etc/os-release; echo $ID$VERSION_ID)

wget https://developer.download.nvidia.com/compute/cuda/repos/${distribution}/x86_64/cuda-keyring_1.1-1_all.deb
sudo dpkg -i cuda-keyring_1.1-1_all.deb
sudo apt update

2. 安装驱动 + CUDA Toolkit（含开发库）

1
sudo apt install -y cuda

这将安装：

  • NVIDIA 驱动（通常是最新版本）
  • cuda-toolkit（包括 nvcc 编译器、头文件、运行时等）
  • 各种开发库（如 libcudart-dev、libcublas-dev 等）

3. 安装 cuDNN（可选，需注册账号）

  1. 登录 https://developer.nvidia.com/cudnn
  2. 下载适用于 Ubuntu 的 .deb 安装包
  3. 安装示例：
1
2
sudo dpkg -i libcudnn8*.deb
sudo apt install -f  # 自动安装依赖

或使用官方仓库（有 CUDA 账号的情况）配置。

4. 设置环境变量

在 ~/.bashrc 或 ~/.zshrc 中添加：

1
2
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH

然后运行：

1
source ~/.bashrc

5. 验证安装

1
2
nvcc --version          # 验证 CUDA 编译器
nvidia-smi              # 查看 GPU 驱动状态

6.常见可选开发库（APT 包名）

库 APT 包名
cuDNN libcudnn8, libcudnn8-dev
NCCL libnccl2, libnccl-dev
TensorRT libnvinfer8, libnvinfer-dev
Thrust 已包含在 cuda-toolkit 中
OpenCL nvidia-opencl-dev, ocl-icd-opencl-dev