0
点赞
收藏
分享

微信扫一扫

安装nvidia-docker的步骤


以下是安装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的容器进行测试。


举报

相关推荐

0 条评论