Install GPU TensorFlow From Sources w/ Ubuntu 16.04 and Cuda 8.0

前端之家收集整理的这篇文章主要介绍了Install GPU TensorFlow From Sources w/ Ubuntu 16.04 and Cuda 8.0前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。

In this tutorial I will be going through the process of building the latest TensorFlow from sources for Ubuntu 16.04. TensorFlow now supports using Cuda 8.0 & CuDNN 5.1 so you can use the pip’s from theirwebsitefor a much easier install. If you would like to install into a Anaconda environment the easiest method is to ‘conda install pip’ and just use the pip packages. If you prefer to build from sources using Ubuntu 14.04 pleasesee my other tutorial.

In order to use TensorFlow with GPU support you must have a Nvidia graphic card with a minimumcompute capabilityof 3.0.

Getting started I am going to assume you know some of thebasics of using a terminalin Linux.


Install required Packages

Open a terminal by pressing Ctrl + Alt + T
Paste each line one at a time (without the $) using Shift + Ctrl + V

$sudoapt-getinstallopenjdk-8-jdkgitpython-devpython3-devpython-numpypython3-numpybuild-essentialpython-pippython3-pippython-virtualenvswigpython-wheellibcurl3-dev


Update & Install Nvidia Drivers

You must also have the 367 (or later) NVidia drivers installed,this can easily be done from Ubuntu’s built in additional drivers after you update your driver packages.

$sudoadd-apt-repositoryppa:graphics-drivers/ppa
$sudoaptupdate

Once installed using additional drivers restart your computer. If you experience any troubles booting linux or logging in: try disabling fast & safe boot in your bios and modifying your grub boot options to enable nomodeset.


Install Nvidia Toolkit 8.0 & CudNN

Skip if not installing with GPU support

To install the Nvidia Toolkit download base installation .run file fromNvidiawebsite.MAKE SURE YOU SAY NO TO INSTALLING NVIDIA DRIVERS!Also make sure you select yes to creating a symbolic link to your cuda directory.

$cd~/Downloads#ordirectorytowhereyoudownloadedfile
$sudoshcuda_8.0.44_linux.run--override#holdstoskip

This will install cuda into:/usr/local/cuda

To install CudNN downloadcudNNv5.1 for Cuda 8.0 from Nvidia website and extract into/usr/local/cudavia:

$sudotar-xzvfcudnn-8.0-linux-x64-v5.1.tgz
$sudocpcuda/include/cudnn.h/usr/local/cuda/include
$sudocpcuda/lib64/libcudnn*/usr/local/cuda/lib64
$sudochmoda+r/usr/local/cuda/include/cudnn.h/usr/local/cuda/lib64/libcudnn*

Then update your bash file:

$gedit~/.bashrc

This will open yourbash filein a text editor which you will scroll to the bottom and add these lines:

exportLD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64"
exportCUDA_HOME=/usr/local/cuda

Once you save and close the text file you can return to your original terminal and type this command to reload your .bashrc file:

$source~/.bashrc

Install Bazel

Instructions also onBazelwebsite

$echo"deb[arch=amd64]http://storage.googleapis.com/bazel-aptstablejdk1.8"|sudotee/etc/apt/sources.list.d/bazel.list$curlhttps://storage.googleapis.com/bazel-apt/doc/apt-key.pub.gpg|sudoapt-keyadd-
$sudoapt-getupdate
$sudoapt-getinstallbazel
$sudoapt-getupgradebazel

Clone TensorFlow

$cd~
$gitclonehttps://github.com/tensorflow/tensorflow


Configure TensorFlow Installation

$cd~/tensorflow
$./configure

Use defaults by pressing enter for all except:

Please specify the location of python. [Default is /usr/bin/python]:

For Python 2 use default or If you wish to build for Python 3 enter:

$/usr/bin/python3.5

Please input the desired Python library path to use. Default is [/usr/local/lib/python2.7/dist-packages]:

For Python 2 use default or If you wish to build for Python 3 enter:

$/usr/local/lib/python3.5/dist-packages

Unless you have a Radeon graphic card you can say no to OpenCL support. (has anyone tested this? ping me if so!)

Please specify the Cuda SDK version you want to use,e.g. 7.0. [Leave empty to use system default]:

$8.0

Please specify the Cudnn version you want to use. [Leave empty to use system default]:

$5

You can find the compute capability of your device at:https://developer.nvidia.com/cuda-gpus

If all was done correctly you should see:

INFO: All external dependencies fetched successfully.
Configuration finished


Build TensorFlow

Warning Resource Intensive I recommend having at least 8GB of computer memory.

If you want to build TensorFlow with GPU support enter:

$bazelbuild-copt--config=cuda//tensorflow/tools/pip_package:build_pip_package

Forcpu onlyenter:

$bazelbuild-copt//tensorflow/tools/pip_package:build_pip_package


Build & Install Pip Package

This will build the pip package required for installing TensorFlow in your /tmp/ folder

$bazel-bin/tensorflow/tools/pip_package/build_pip_package/tmp/tensorflow_pkg

To Install Using Python 3 (remove sudo if using a virtualenv)

$sudopip3install/tmp/tensorflow_pkg/tensorflow

#withnospacesaftertensorflowhittabbeforehittingentertofillinblanks

For Python 2 (remove sudo if using a virtualenv)

$sudopipinstall/tmp/tensorflow_pkg/tensorflow

#withnospacesaftertensorflowhittabbeforehittingentertofillinblanks


Test Your Installation

Close all your terminals and open a new terminal to test.

$python#orpython3
$importtensorflowastf
$sess=tf.InteractiveSession()
$sess.close()

TensorFlow also hasinstructionson how to do a basic test and a list of common installation problems.

There you have it,you should now have TensorFlow installed on your computer. This tutorial was tested on a fresh install of Ubuntu 16.04 with a GeForce GTX 780 and a GTX 970m.

If you want to give your GPU a workout maybe try building a massive image classifier following thistutorial.

https://alliseesolutions.wordpress.com/2016/09/08/install-gpu-tensorflow-from-sources-w-ubuntu-16-04-and-cuda-8-0-rc/

原文链接:https://www.f2er.com/ubuntu/355002.html

猜你在找的Ubuntu相关文章