Ubuntu18.04下Tensorflow1.9环境搭建

本文讲解直接将Tensorflow-gpu1.9版本安装于Anaconda的base环境中,也就是系统默认环境
查询Tensorflow1.9配合的环境为Cuda9.0和Cudnn7.1.2~7.1.4
前提:已经装好Nvidia显卡驱动


1.安装Cuda9.0

首先安装依赖库

sudo apt-get install freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev

在官网上下载run 文件,如图所示,选择合适run文件
这里写图片描述
CUDA9.0要求GCC版本是5.x或者6.x,需要自己修改版本,先查看gcc版本

$ g++ --version  

这里写图片描述
安装gcc5

$ sudo apt-get install gcc-5
$ sudo apt-get install g++-5

替换掉之前的版本:

$ sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-5 50
$ sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-5 50

检查是否替换成功

$ g++ --version  

在文件所在处启动终端,运行cuda.run文件

$ sudo sh cuda_9.0.176_384.81_linux.run

在安装过程中会提示是否需要安装显卡驱动,如图所示,在这里要选择n,其他的选择y或者回车键进行安装
这里写图片描述
安装完成如下图
这里写图片描述
进行环境配置

$ sudo gedit ~/.bashrc

会弹出一个可写的配置文件,在末尾把以下配置写入并保存

export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}}  
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

最后执行

$ source ~/.bashrc

之后重启
在安装的时候也也相应安装了一些cuda的一些例子,可以进入例子的文件夹然后使用make命令执行
例一:

$ cd /usr/local/cuda-9.0/samples/1_Utilities/deviceQuery # 第一步,进入例子文件
$ sudo make # 第二步,执行make命令
$ ./deviceQuery  # 第三步

如果结果有GPU的信息,说明安装成功。
例二:

$ cd NVIDIA_CUDA-9.0_Samples/5_Simulations/fluidsGL # 进入例子对应的文件夹
$ make clean && make # 执行make
$ ./fluidsGL # 运行

当运行这个程序的时候我们可以通过如下命令查看GPU是否在使用:

nvidia-smi

可以看到我的1070MQ占用率99%
这里写图片描述
停止运行后占用率为0
这里写图片描述

2.安装cuDnn7.1.4

7.1.4和7.1.2与我们的环境都是匹配的
首先到官网下载,可能需要注册
这里写图片描述
下载其中的
cuDNN v7.1.4 Runtime Library for Ubuntu16.04 (Deb)
cuDNN v7.1.4 Developer Library for Ubuntu16.04 (Deb)
cuDNN v7.1.4 Code Samples and User Guide for Ubuntu16.04 (Deb)
下载完成后,在文件所在目录下按顺序安装

$ sudo dpkg -i libcudnn7_7.0.3.11-1+cuda9.0_amd64.deb
$ sudo dpkg -i libcudnn7-dev_7.0.3.11-1+cuda9.0_amd64.deb
$ sudo dpkg -i libcudnn7-doc_7.0.3.11-1+cuda9.0_amd64.deb

测试安装是否成功,

$ cp -r /usr/src/cudnn_samples_v7/ $HOME
$ cd $HOME/cudnn_samples_v7/mnistCUDNN
$ make clean && make
$ ./mnistCUDNN

出现“Test passed! ”,则说明安装成功

3.Anaconda安装

官网下载
在文件路径下

$ bash Anaconda3-5.2.0-Linux-x86_64.sh

安装完成后询问是否添加环境变量,选择y‘
这样当调用python时,就会调用Anaconda默认环境base中的python

4.pycharm的安装

官网下载
注册账户
如果有教育网邮箱可以申请免费用,每年重新申请一次
解压缩,复制到想要安装的路径下,运行bin文件夹下的sh文件安装
安装完成后可以创建快捷方式:

tools-在桌面创建快捷方式
创建新工程时,编译器选择为Anaconda下的诸环境(包括默认环境)下想要的环境中的python3.6

5.Tensorflow-gpu的安装

安装指定版本的tensorflow-gpu(默认安装最新版本)

$ pip install tensorflow-gpu==1.9.0 
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值