Ubuntu18.04 安裝 docker 和 NVIDIA Container Toolkit（使用外接顯卡）

顯卡規格：Nvidia 1070 Ti

顯卡驅動版本：Driver Version: 390.116

nvidia-smi -l

確認Ubuntu 有抓到顯卡驅動後就可以開始來安裝docker and nvidia-docker

安裝版本與步驟參考Nvidia github wiki

NVIDIA/nvidia-docker

第一步 請安裝Docker-ce，也就是Community版本，使用docker 19.03

Make sure you have installed the NVIDIA driver and Docker 19.03 for your Linux distribution

安裝步驟參考

Get Docker Engine - Community for Ubuntu

現在新版的Docker以支持原生GPU顯卡，不需使用 nvidia-docker2 packages

現在統一正名為NVIDIA Container Toolkit

For first-time users of Docker 19.03 and GPUs, continue with the instructions for getting started below.

設定方式如下：

# 新增套件＆系統更新&安裝
$ 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

驗證 docker 是否可以抓到顯卡

＃Test nvidia-smi with the latest official CUDA image
$sudo docker run --gpus all nvidia/cuda:9.0-base nvidia-smi

正常可以在docker中抓到顯卡使用cuda 9.0

接著設定一個docker 可遠端桌面且有GPU版本的系統環境

使用ubuntu-desktop-lxde-vnc Image

Docker Hub

＃下載ubuntu-desktop-lxde-vnc
$sudo docker pull dorowu/ubuntu-desktop-lxde-vnc

新增建立一個有GPU 的IMG Container

$sudo docker run --gpus all  -p 6080:80 -v /dev/shm:/dev/shm dorowu/ubuntu-desktop-lxde-vnc

無GPU的版本

$sudo docker run -p 6081:80 -v /dev/shm:/dev/shm dorowu/ubuntu-desktop-lxde-vnc

使用瀏覽器VNC 進入遠端桌面

http://127.0.0.1:6080

GPU版本Container

這樣就成功建立起有GPU環境的Containe

可以在裡面安裝各種所需要的開發工具～

Grady Huang