Ubuntu docker-NVIDIA 支持显卡调用

1. Ubuntu系统安装

下载Ubuntu操作系统并安装，下载地址: https://ubuntu.com/#download , 安装操作系统步骤请自行百度。

2. 显卡驱动安装

安装显卡驱动请参照链接: Ubuntu16.04 NVIDIA显卡驱动卸载与安装

3. docker安装与卸载

3.1 卸载docker

卸载docker请参考文章: Ubuntu 16.04 Docker卸载

3.2. 安装docker

• 在线安装

  • 1、 安装必要的一些系统工具

    sudo apt-get update
    sudo apt-get -y install apt-transport-https ca-certificates curl software-properties-common
    

  • 2、 安装GPG证书

    curl -fsSL http://mirrors.aliyun.com/docker-ce/linux/ubuntu/gpg | sudo apt-key add -
    

  • 3、写入软件源信息

    sudo add-apt-repository "deb [arch=amd64] http://mirrors.aliyun.com/docker-ce/linux/ubuntu $(lsb_release -cs) stable"
    

  • 4、更新并安装 Docker-CE

    sudo apt-get -y update
    apt-cache madison docker-ce
    sudo apt-get -y install docker-ce=18.03.1~ce-0~ubuntu
    

• 离线安装

  • 1、 下载离线安装程序，下载地址: docker版本下载, 可自行下载不同版本的docker安装程序，总共下载三个文件，如图：
    
    我这里提供19.03.5的打包版本地址
    链接： docker 19.03.5下载
    提取码：C50H

  • 2、 将下载文件解压并将文件夹下的deb文件上传到服务器，并进入到目录下执行：

    dpkg -i *.deb
    

  • 3、配置docker仓库源地址，编辑/etc/docker/daemon.json文件, 添加以下内容：

    {
    "registry-mirrors": ["https://3c9ywpon.mirror.aliyuncs.com"],
    "insecure-registries": ["192.168.4.125:5000"]
    } 
    

    registry-mirrors： 表示公有仓库地址
    insecure-registries： 私有仓库地址
    根据需要进行修改， 修改后保存。

  • 4、重新加载配置与重启docker，执行以下两条命令：

    systemctl daemon-reload 
    systemctl restart docker
    

    输入命令: docker -v 返回docker版本信息则按照成功

4、NVIDIA-Docker安装

1、下载nvidia-container-toolkit安装插件，下载链接：
链接： 插件下载
提取码：C50H
2、将下载的文件解压并放到安装docker服务器上，进入到目录下执行：

dpkg -i *.deb

3、重启docker服务：

systemctl restart docker

4、测试验证docker容器是否可正常调用GPU显卡

docker run --gpus all --rm nvidia/cuda:9.0-base nvidia-smi

如果正确返回NVIDIA显卡相关信息则证明安装成功，后期就执行替换镜像名称来启动你自己需要的容器服务。

5、总结

以上的操作步骤，是本人在工作中亲测可用，也走过很多弯路，写这篇文章也是因为在群里有朋友再询问docker如何调用NVIDIA驱动，所以为了方便自己以后工作中更方便的查找记录，顺便也发布出来和大家一起分享，觉得本文不错的朋友，记得收藏关注一下！