Ubuntu 16.04 安装深度学习环境 GPU 加速版

深度学习环境搭建起来比较麻烦，特别是 GPU 加速版。既要安装 Nvidia 的驱动，又要安装 cuda 和 cuDNN 。本文通过 docker 容器技术搭建基于keras (tensorflow) 的深度学习环境。

Docker 是一种容器技术，将你要运行的任何环境只通过几行命令就可以搭建完成。而 Docker 中除了用普通的 docker pull 方式直接拉去现成的深度学习镜像来搭建深学环境外，还可以通过用 Dockerfile 来完成这一任务。

经过笔者的不（jian）懈（chi）努（cai）力（keng），一个完整的经过测试的深度学习 Dockerfile 见以下链接：

https://github.com/zhudaoruyi/deep-learning-gpu-env

从 GitHub 上 clone 完后，（记住给个 Star ）进入到 Dockerfile 所在的目录。

Dockerfile 构建镜像的方法

docker build -t name:tag .

运行该镜像的方法gpu加速)运行该镜像的方法(GPU加速)

nvidia-docker run -d -p 8888:8888 --name test -v /home/pzw:/home/workspace 镜像ID
nvidia-docker exec -it 容器ID

保存该镜像的方法)保存该镜像的方法

docker save -o 镜像名称

保存该容器的方法

docker export -o 容器名称

---

nvidia-docker 的安装

参考 https://github.com/zhudaoruyi/nvidia-docker

注意：安装 nvidia-docker 之前先安装好 docker

为了确认 nvidia-docker 是否安装成功，运行

nvidia-docker run --rm nvidia/cuda nvidia-smi

如果正确输出了本机的 GPU 信息，则安装成功。
例如：

+-----------------------------------------------------------------------------+
| NVIDIA-SMI 367.48                 Driver Version: 367.48                    |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|===============================+======================+======================|
|   0  Tesla M40 24GB      Off  | 0000:02:00.0     Off |                    0 |
| N/A   33C    P0    57W / 250W |  22427MiB / 22939MiB |      0%      Default |
+-------------------------------+----------------------+----------------------+
|   1  Tesla M40 24GB      Off  | 0000:82:00.0     Off |                    0 |
| N/A   37C    P0    58W / 250W |  21663MiB / 22939MiB |      0%      Default |
+-------------------------------+----------------------+---------------------