以下是安装nvidia-docker的步骤:
- 安装Docker CE
首先需要在节点上安装Docker CE。可以使用官方教程进行安装。 - 安装NVIDIA GPU驱动
在Kubernetes节点上安装NVIDIA GPU驱动,可以使用NVIDIA提供的官方安装脚本。 - 添加NVIDIA repository
添加NVIDIA repository,以便能够下载nvidia-docker的安装包。使用以下命令添加repository:
curl -s -L https://nvidia.github.io/nvidia-docker/ubuntu18.04/nvidia-docker.list | \
sudo tee /etc/apt/sources.list.d/nvidia-docker.list
这里以Ubuntu18.04为例。
- 安装nvidia-docker
使用以下命令安装nvidia-docker:
sudo apt-get update && sudo apt-get install -y nvidia-docker2
安装完成后,需要重启Docker进程:
sudo systemctl restart docker
- 验证安装
使用以下命令验证nvidia-docker是否安装成功:
sudo docker run --rm nvidia/cuda:9.0-base nvidia-smi
输出应该显示正在使用NVIDIA GPU。
至此,nvidia-docker的安装就完成了。可以在Kubernetes中部署带有GPU的容器进行测试。