Accéder au contenu principal

Installing Tensorflow with GPU support on Ubuntu 16.04



Installing Docker for running learning resources (eg Udacity MOOC)

1. Head to the Docker installation webpage and install Docker https://docs.docker.com/engine/installation/linux/ubuntulinux/

2. If you have a large second HDD, you might want to make sure all the docker images you use are stored on the large disk to preserve your precious SSD space.
For this, I found that creating a symbolic link to a new location on the second hard-drive disk worked best for me.

  a. Make sure that the large hard drive disk is mounted automatically when booting the disk
https://help.ubuntu.com/community/Fstab

  b. Setup a symbolic link to a directory inside your large hard drive disk. Setting DOCKER_OPTS did not work for me, but the symbolic link method worked:
https://forums.docker.com/t/how-do-i-change-the-docker-image-installation-directory/1169


Installing Python Anaconda

1. Follow the instructions here https://www.continuum.io/downloads

2. Download this python script (It will update the PIP libraries useful to download and easily install Tensorflow later-on)  https://bootstrap.pypa.io/ez_setup.py

3. Run the script in a Bash Terminal "python ez_setup.py "


Installing the GPU computing libraries


1. Install CUDA 8.0 following these instructions
http://askubuntu.com/questions/799184/how-can-i-install-cuda-on-ubuntu-16-04

2. At the end of the process run the command "ls /usr/local/cuda/" to check that the installation has been successfully completed. You should see the following directories: "include" and "lib64"

3. Restart your machine

4. To install the NVIDIA CuDNN v5 library follow these instructions (it is mostly about creating an NVIDIA dev account, downloading an archive and copying the files inside ls /usr/local/cuda/"):
http://askubuntu.com/questions/767269/how-can-i-install-cudnn-on-ubuntu-16-04

5. Add the libraries to your environment variable PATH by adding at the end of the ".bashrc" file:

# Added support for CUDA and CUDnn (NVIDIA GPU-accelerated library of primitives for deep neural networks)
LD_LIBRARY_PATH="/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64"
export CUDA_HOME="/usr/local/cuda"
export LD_LIBRARY_PATH
export PATH="$LD_LIBRARY_PATH:$PATH"


Installing Tensorflow with GPU support


To install Tensorflow, run the following:
# Ubuntu/Linux 64-bit, GPU enabled, Python 3.5
# Requires CUDA toolkit 8.0 and CuDNN v5. For other versions, see "Installing from sources" below.
$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-0.12.0rc0-cp35-cp35m-linux_x86_64.whl
$ sudo pip install --upgrade $TF_BINARY_URL


To test your installation of Tensorflow, run the following in a Bash Terminal
$ python
...
>>> import tensorflow as tf
>>> hello = tf.constant('Hello, TensorFlow!')
>>> sess = tf.Session()
>>> print(sess.run(hello))
Hello, TensorFlow!
>>> a = tf.constant(10)
>>> b = tf.constant(32)
>>> print(sess.run(a + b))
42
>>>





Commentaires

Unknown a dit…
Thanks a million! I just got the Alienware R6 and try to install Ubuntu 16.04, but every time when I shut down the computer there is a kernel panic exactly the same. This problem has bothered me for days, and I have reinstalled the system several times. God I finally find your page. Really thanks a lot, do you know what is exactly the reason for this kernel panic? I think it is problem of Dell, because I never encounter this problem in my laptop.

Posts les plus consultés de ce blog

Setting up Rust on Windows 10 with MSYS2 without installing Visual Studio

Setting up Rust on Windows 10 with MSYS2 without installing Visual Studio  1. Install MSYS2 2. Update MSYS2 and run : pacman -Sy && pacman -Syu  and install git and gcc: pacman -S git &&  pacman -S mingw-w64-x86_64-toolchain 3. Open the MSYS2 terminal and install the rust compiler with :  pacman -S mingw-w64-x86_64-rust You may stop here but you will miss the rustup package (which is not part of MSYS2) to manage the Rust compiler environment. To install rustup , follow the next steps. 4. Clone the rustup git repository :  git clone git@github.com:rust-lang/rustup.git --depth=1 5. Compile and run rustup: cd rustup cargo run --release 6. Once rustup detects the previously installed rust compiler and asks to uninstall it before you can continue. So, open a new MSYS2 terminal and uninstall rust :  pacman -R mingw-w64-x86_64-rust 7. Now back to the rustup terminal, choose yes to continue the installation. On the next question choose : custom installation and chose as host t

Installing Ubuntu 16.04 on an Alienware Aurora R5

Installing Ubuntu 16.04 LTS on an Alienware Aurora R5 with NVIDIA GPU support (Dual boot Windows) I have compiled these instructions from my succesful installation on Nov 25. Would you be happy to live without Windows? If not use Dual Boot. This procedure is inspired from this DELL support page : http://www.dell.com/support/article/uk/en/ukbsdt1/SLN301754/en You can refer to the link above for additional steps and to see useful screenshots. However, please, make sure to take care about the GPU-related  steps below. Re-installing Windows 10 with BIOS Secure Boot to OFF First, check that have only one disk installed in your machine. I have found that having one disk only makes it easier to re-install Windows. Secondly, an important step is to create two USB Recovery disks. Use two USB memory sticks bigger than 16 GB supporting the USB 3.0 standard (file copy will be much faster). Use the first one to create a recovery disk for Windows 10, as a backup if the installati