AI Cloud Hook

Appearance

Docker 安装 CUDA ​

记录在 Ubuntu 上安装 CUDA 环境的步骤。

前置条件 ​

  • Ubuntu 20.04/22.04
  • NVIDIA 显卡驱动已安装

安装步骤 ​

bash
# 1. 添加 NVIDIA 仓库
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dearmour -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg
curl -s -L https://nvidia.github.io/libnvidia-container/$distribution/libnvidia-container.list | sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' | sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list

# 2. 安装 CUDA Toolkit
sudo apt-get update
sudo apt-get install -y cuda-toolkit-12-4

# 3. 验证
nvcc --version

Docker 配置 ​

bash
# 安装 nvidia-container-toolkit
sudo apt-get install -y nvidia-container-toolkit

# 配置 Docker 使用 NVIDIA 运行时
sudo nvidia-ctk --version

测试 ​

bash
docker run --rm --gpus all nvidia/cuda:12.4.0-base nvidia-smi