Docker: 小白之路六（Docker使用篇之Nvidia-docker）

Docker:小白之路六（Docker使用篇之Nvidia-docker）

1. NVIDIA Container Toolkit 介绍

NVIDIA Container Toolkit允许用户构建和运行GPU加速的Docker容器。 该工具包包括容器运行时库和实用程序，用于自动配置容器以利用NVIDIA GPU。 完整的文档和常见问题解答可在资源库Wiki上获得。

2. Quick start

确保已为Linux发行版安装了NVIDIA驱动程序和Docker 19.03。请注意，您无需在主机上安装CUDA工具包，但需要安装驱动程序

请注意，随着Docker 19.03的发布，不赞成使用nvidia-docker2软件包，因为Docker运行时中现在已将NVIDIA GPU作为设备本地支持。

查看一下自己docker的版本

root@felaim-pc:/home/felaim/Documents/docker_tutorial# docker --version
Docker version 19.03.8, build afacb8b7f0

对于Docker 19.03和GPU的初学者，请继续以下入门说明(LZ今天刚接触docker，妥妥的初学小白)

Ubuntu 16.04/18.04, Debian Jessie/Stretch/Buster

# Add the package repositories
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list

sudo apt-get update && sudo apt-get install -y nvidia-container-toolkit
sudo systemctl restart docker

测试一下：

#### Test nvidia-smi with the latest official CUDA image
docker run --gpus all nvidia/cuda:10.0-base nvidia-smi

能够看到LZ笔记本的小卡信息，就表示安装成功啦！

参考地址：

1.https://github.com/NVIDIA/nvidia-docker