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 地址

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

扫码关注云+社区

领取腾讯云代金券