Ubuntu 18.04 Server 搭建tensorflow docker运行环境

一、系统安装

参考ubuntu-server系统安装

二、显卡驱动安装

显卡型号：Nvidia GTX1050.

1、首先去Nvidia官网下载驱动：

image

找到自己的显卡，选择Linux版本进行下载，下载的是.run文件。

2、禁用nouveau

安装NVIDIA需要把系统自带的驱动禁用，打开文件：

sudo gedit /etc/modprobe.d/blacklist.conf

在文本最后添加以下内容：

blacklist nouveau
option nouveau modeset=0

命令窗口会提示warn，无视之。

保存退出，执行以下命令生效：

sudo update-initramfs -u

重启电脑后输入：

lsmod | grep nouveau

没有任何输出说明禁用成功。

3、安装NVIDIA显卡驱动

在安装驱动程序的过程中，会因为缺少gcc、g++、make等development tool而报错导致无法完成驱动程序安装（ERROR：Ubable to find the development tool 'make' in your path...），不用担心，手动安装这些开发包后再次执行安装指令即可。另外，安装过程中出现的弹框根据默认选项选择即可。手动安装development tool指令为:

sudo apt-get install gcc
sudo apt-get install g++
sudo apt-get install make

4、 查看显卡驱动版本

可以通过nvidia-smi命令查看GPU和驱动程序信息，若出现以下结果，则表明驱动程序安装成功。

安装时没有截图，所以用了网上的图片

三、安装docker

1、安装基础docker服务

安装基础的docker请参考Ubuntu下通过docker部署springboot项目 安装和配置docker部分。

2、安装nvidia-docker2服务

想要在docker容器中使用宿主机的gpu，则需要安装一个nvidia-docker的插件。下面介绍安装方法：
第一种：

wget -P /tmp https://github.com/NVIDIA/nvidia-docker/releases/download/v1.0.1/nvidia-docker_1.0.1-1_amd64.deb
sudo dpkg -i /tmp/nvidia-docker*.deb

第二种：

sudo apt-get install nvidia-docker2

本人通过第二种方式安装，第一种方式未经测试。
可以直接去nvidia-docker 的github看看：https://github.com/NVIDIA/nvidia-docker/

3、测试nvidia-docker

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

然后在执行下面这句，默认用nvdia-docker替代docker命令:

echo 'alias docker=nvidia-docker' >> ~/.bashrc
bash

最后就可以下载tensorflow镜像进行测试了，这里就不详细介绍了。
附赠 tensorflow docker hub 地址