1. 安装 NVIDIA 驱动程序和 CUDA Toolkit
确保你的系统上安装了 NVIDIA GPU 驱动程序,并且版本与 Docker 期望的版本兼容。
- 运行
nvidia-smi
命令来检查 NVIDIA 驱动程序是否正确安装。
2. 安装 nvidia-container-toolkit
对于 Docker 版本 19.03 及以上,使用 nvidia-container-toolkit
替代旧的 nvidia-docker2
。
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-do