前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >Ubuntu 18.04 Server 搭建tensorflow docker运行环境

Ubuntu 18.04 Server 搭建tensorflow docker运行环境

作者头像
吟风者
发布2019-09-18 11:43:42
1.1K0
发布2019-09-18 11:43:42
举报
文章被收录于专栏:吟风者

一、系统安装

参考ubuntu-server系统安装

二、显卡驱动安装

显卡型号:Nvidia GTX1050.

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

image

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

2、禁用nouveau

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

代码语言:javascript
复制
sudo gedit /etc/modprobe.d/blacklist.conf

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

代码语言:javascript
复制
blacklist nouveau
option nouveau modeset=0

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

保存退出,执行以下命令生效:

代码语言:javascript
复制
sudo update-initramfs -u

重启电脑后输入:

代码语言:javascript
复制
lsmod | grep nouveau

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

3、安装NVIDIA显卡驱动

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

代码语言:javascript
复制
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的插件。下面介绍安装方法:

第一种:

代码语言:javascript
复制
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

第二种:

代码语言:javascript
复制
sudo apt-get install nvidia-docker2

本人通过第二种方式安装,第一种方式未经测试。

可以直接去nvidia-docker 的github看看:https://github.com/NVIDIA/nvidia-docker/

3、测试nvidia-docker
代码语言:javascript
复制
nvidia-docker run --rm nvidia/cuda nvidia-smi

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

代码语言:javascript
复制
echo 'alias docker=nvidia-docker' >> ~/.bashrc
bash

最后就可以下载tensorflow镜像进行测试了,这里就不详细介绍了。

附赠 tensorflow docker hub 地址

本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2019.09.16 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 一、系统安装
  • 二、显卡驱动安装
    • 1、首先去Nvidia官网下载驱动:
      • 2、禁用nouveau
        • 3、安装NVIDIA显卡驱动
          • 4、 查看显卡驱动版本
          • 三、安装docker
            • 1、安装基础docker服务
              • 2、安装nvidia-docker2服务
                • 3、测试nvidia-docker
                相关产品与服务
                容器镜像服务
                容器镜像服务(Tencent Container Registry,TCR)为您提供安全独享、高性能的容器镜像托管分发服务。您可同时在全球多个地域创建独享实例,以实现容器镜像的就近拉取,降低拉取时间,节约带宽成本。TCR 提供细颗粒度的权限管理及访问控制,保障您的数据安全。
                领券
                问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档