artifacts/docs/gpu-containerd.md

在 Ubuntu 上安装 NVIDIA 驱动和 nvidia-container-toolkit

以下步骤演示如何在主机安装 NVIDIA 驱动及 nvidia-container-toolkit，并将 containerd 配置为能够使用 GPU。末尾还补充了 sealos 自带 containerd 的配置方式。

1. 安装 NVIDIA 驱动和 nvidia-container-toolkit

# 添加 NVIDIA 容器工具箱仓库
sudo apt-get update
sudo apt-get install -y ca-certificates curl gnupg
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | \
  sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg

distribution=$(awk -F= '/^ID=/{print $2}' /etc/os-release)$(awk -F= '/^VERSION_ID=/{print $2}' /etc/os-release)
curl -s -L "https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list" | \
  sed 's#^deb #deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] #' | \
  sudo tee /etc/apt/sources.list.d/nvidia-docker.list

sudo apt-get update
sudo apt-get install -y nvidia-driver-535 nvidia-container-toolkit

安装完成后可通过 nvidia-smi 验证驱动是否正常：

nvidia-smi

2. 配置 containerd 使用 GPU

使用 nvidia-ctk 工具可以快速生成配置并设置为默认运行时：

sudo nvidia-ctk runtime configure --runtime=containerd --set-as-default
sudo systemctl restart containerd

以上命令会在 /etc/containerd/config.toml 中新增 nvidia 运行时，使 kubelet 或其他工具可以直接调度 GPU 容器。

3. sealos-containerd 支持

若主机通过 sealos 部署集群，其内置的 containerd 服务名通常为 sealos-containerd，配置文件位于 sealos 数据目录，例如： /var/lib/sealos/data/default/rootfs/etc/containerd/config.toml。可按以下方式配置：

sudo nvidia-ctk runtime configure \
  --config /var/lib/sealos/data/default/rootfs/etc/containerd/config.toml \
  --set-as-default
sudo systemctl restart sealos-containerd

完成后即可在 sealos 集群内运行需要 GPU 的容器或 Pod。