NVIDIA Container Toolkit 安装

NVIDIA Container Toolkit是一种工具包，它允许用户在Docker容器中运行GPU加速的 workload。安装步骤通常包括以下几个部分：

1. **更新系统**：首先确保你的主机系统是最新的，因为某些依赖项可能会需要。

sudo apt-get update && sudo apt-get upgrade

2. **安装NVIDIA Docker**: 使用官方仓库安装NVIDIA Docker，这会自动安装必要的库和驱动支持。

   curl -s https://nvidia.github.io/nvidia-docker/gpgkey | gpg --dearmor -o /usr/local/share/keyrings/nvidia-archive-keyring.gpg
   echo "deb [signed-by=/usr/local/share/keyrings/nvidia-archive-keyring.gpg] http://us.download.nvidia.com/repos/ubuntu/xenial/compute-54/ $UBUNTU_CODENAME main" | tee /etc/apt/sources.list.d/nvidia-docker.list
   sudo apt-get update
   sudo apt-get install nvidia-container-toolkit

3. **验证安装**：安装完成后，你可以通过`docker run`命令测试是否能使用GPU资源，比如运行一个简单的TensorFlow镜像。

   docker run --gpus all nvidia/cuda:latest nvidia-smi