ubuntu14.04系统中安装tensorflow(gpu版)

前端之家收集整理的这篇文章主要介绍了ubuntu14.04系统中安装tensorflow(gpu版)前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。
原文:http://blog.csdn.net/u012969412/article/details/64502393

系统配置:ubuntu14.04 + GTX1060 + i7(7700K)

安装部署参考地址1:http://www.jb51.cc/article/p-uhgocanx-vb.html

安装部署参考地址2(推荐):http://blog.csdn.net/zhaoyu106/article/details/52793183

安装前准备工作(离线安装)

官方下载:cuda-8.0 toolkit

github下载tensorflow*.whl文件tensorflow_GPU_py2.7 (拉到最下面选择GPU+py2)

官方下载:cuDNNV5(需要注册登录官网)

一共3个文件放入U盘,在ubuntu实体机上读取安装。这些文件全部放在/usr/local/WYLdownload目录下

第一步:安装nvidia显卡驱动

Linux用户可以通过官方ppa解决安装GPU驱动的问题。使用如下命令添加Graphic Drivers PPA:

  1. sudoadd-apt-repositoryppa:graphics-drivers/ppa
然后更新源:
  1. sudoapt-getupdate
然后去navidia官网查看最新的驱动版本号:navidia官网: http://www.geforce.cn/drivers
比如说驱动的最新版本号为375,则执行如下指令:
  1. sudoapt-getinstallnvidia-375
最后安装openGL支持
  1. sudoapt-getinstallmesa-common-dev

第二步:安装cuda-toolkit-8.0并用其安装cuda-8.0

如果直接执行:$ sudoapt-getinstall-ycuda 会报错。正确使用方法为。
官方网站下载:CUDA Toolkit 8.0。下载网址为: https://developer.nvidia.com/cuda-downloads

后执行如下指令:
  1. $sudodpkg-icuda-repo-ubuntu1404-8-0-local-ga2_8.0.61-1_amd64.deb
  2. $sudoapt-getupdate
  3. $sudoapt-getinstallcuda
这时在/usr/local目录下产生一个cuda安装的路径叫"cuda-8.0"添加cuda到环境变量:
  1. sudovim/etc/profile
添加内容
  1. exportCUDA_HOME=/usr/local/cuda-8.0
  2. exportPATH=$CUDA_HOME/bin:$PATH
  3. exportLD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH
使环境变量生效
  1. source/etc/profile
检验是否安装成功:查看GPU运行的进程
  1. $nvidia-smi

下载cuda测试用例:下载到~/cuda_examples目录下

  1. $cuda-install-samples-8.0.sh~/cuda_examples
运行测试用例:
  1. $cd~/cuda_examples/NVIDIA_CUDA-8.0_Samples
  2. $make

第三步:降低gcc版本到5.0以下

查看gcc当前使用版本:
  1. $gcc--help
  1. $gcc--version#查看gcc版本号
最后一行为 <file:///usr/share/doc/gcc-4.8/README.Bugs>. 使用的ubuntu14.04使用的是4.8版本 所以不用降低gcc版本

否则执行如下指令:
  1. $sudoapt-getinstallg++-4.9
  2. $sudoupdate-alternatives--install/usr/bin/gccgcc/usr/bin/gcc-4.920
  3. $sudoupdate-alternatives--install/usr/bin/gccgcc/usr/bin/gcc-510
  4. $sudoupdate-alternatives--install/usr/bin/g++g++/usr/bin/g++-4.920
  5. $sudoupdate-alternatives--install/usr/bin/g++g++/usr/bin/g++-510
  6. $sudoupdate-alternatives--install/usr/bin/cccc/usr/bin/gcc30
  7. $sudoupdate-alternatives--setcc/usr/bin/gcc
  8. $sudoupdate-alternatives--install/usr/bin/c++c++/usr/bin/g++30
  9. $sudoupdate-alternatives--setc++/usr/bin/g++

第四步:下载 cuDNN V5 库文件添加到cuda-8.0库

到官网下载:cudnn-7.0-linux-x64-v3.0.8-prod.tgz

解压并将内容copy到/usr/local/cuda-8.0/include和lib64目录中:

  1. $sudotarxvzfcudnn-8.0-linux-x64-v5.1.tgz
  2. $sudocpcuda/include/cudnn.h/usr/local/cuda-8.0/include
  3. $sudocpcuda/lib64/libcudnn*/usr/local/cuda-8.0/lib64
  4. $sudochmoda+r/usr/local/cuda-8.0/include/cudnn.h/usr/local/cuda-8.0/lib64/libcudnn*#分配包的权限

第五步:安装tensorflow

一、Anaconda安装tensorflow(不推荐conda目录和系统自带python目录冲突):

官方安装tensorflow说明:https://www.tensorflow.org/install

1、Create a conda environment named tensorflow to run a version of Python by invoking the following command:建立tensorflow运行环境

  1. $condacreate-ntensorflow

2、Activate the conda environment by issuing the following command:激活conda环境
  1. $sourceactivatetensorflow
3、Issue a command of the following format to install TensorFlow inside your conda environment:

$ pip install --ignore-installed --upgrade TF_PYTHON_URL
其中TF_PYTHON_URL是想要配置的tensorflow版本:
如:https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-1.0.1-cp27-none-linux_x86_64.whl

  1. $sudopipinstall--ignore-installed--upgradehttps://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-1.0.1-cp27-none-linux_x86_64.whl

4、从conda环境中退出

  1. $sourcedeactivate

二、git安装tensorflow

1、克隆Tensorflow仓库
#安转Git支持:sudo apt-get install git

  1. $sudogitclone--recurse-submoduleshttps://github.com/tensorflow/tensorflow/usr/local/WYLdownload/tensorflow
  2. $pipinstall--upgradesetuptoolspip
git clone <版本库的网址> <本地目录名>
即:上述指令是将git上的tensorflow包下载到ubuntu系统的/usr/local/WYLdownload/tensorflow包下
2、配置configure参数
  1. $sudo./configure

三、pip安装tensoflow

文件目录下执行:
  1. $sudopipinstall--ignore-installed--upgradetensorflow_gpu-1.0.1-cp27-none-linux_x86_64.whl

四、Bazel安装tensorflow(如果选择用Bazel安装tensorfloe***.whl)

Bazel是一个类似于Make的工具,是Google为其内部软件开发的特点量身定制的工具,如今Google使用它来构建内部大多数的软件。
但是bazel需要的jdk非以上jdk。需要oracle自己的jdk8包。
  1. $sudoadd-apt-repositoryppa:webupd8team/java
  2. $sudoapt-getupdate
  3. $sudoapt-getinstalloracle-java8-installer
2、安装Bazel依赖:
  1. $echo"deb[arch=amd64]http://storage.googleapis.com/bazel-aptstablejdk1.8"|sudotee/etc/apt/sources.list.d/bazel.list
  2. $curlhttps://bazel.build/bazel-release.pub.gpg|sudoapt-keyadd-
  3. $sudoapt-getinstallpython-numpyswigpython-devpython-wheel
3、安装源支持与bazel并更新bazel:
  1. $sudoapt-getupdate
  2. $sudoapt-getinstallbazel
  3. $sudoapt-getupgradebazel

执行如下指令查看bazel是否安装完成:

  1. $bazelversion
往后略可以看官网

猜你在找的Ubuntu相关文章