Tensorflow

TensorFlow is an open source library for deep learning.

Installation instructions

System-wide CUDA install

Detailed instructions are provided on the TensorFlow website.

1. Install CUDA
   1. Download CUDA from the nVidia site (CUDA toolkit 7.5):

      wget http://developer.download.nvidia.com/compute/cuda/7.5/Prod/local_installers/cuda-repo-ubuntu1504-7-5-local_7.5-18_amd64.deb

   2. Update root path (if necessary):

      sudo nano /root/.bashrc

      export PATH=$PATH:/usr/local/sbin:/usr/sbin:/sbin:/opt/conda/bin

   3. Install deb (c.f. instructions here):

      sudo -s

      dpkg -i cuda-repo-ubuntu1504-7-5-local_7.5-18_amd64.deb

      apt-get update

      aptitude install cuda

2. Install CUDA (alternate method, if above doesn't work):
   1. Download CUDA runfile from the nVida site (CUDA toolkit 7.5 runfile):

      wget http://developer.download.nvidia.com/compute/cuda/7.5/Prod/local_installers/cuda_7.5.18_linux.run

   2. Disable nouveau:

      sudo nano /etc/modprobe.d/blacklist-nouveau.conf

      blacklist nouveau

      options nouveau modeset=0

      sudo -s

      update-initramfs -u

   3. Install

      chmod +x cuda_7.5.18_linux.run

      sudo ./cuda_7.5.18_linux.run

3. Download cuDNN

Virtualenv method

1. Set up environment

   1. nano ~/.bashrc

      # Add these lines to the end of your .bashrc file:

      export CUDA_HOME=/usr/local/cuda-7.5

      export LD_LIBRARY_PATH=${CUDA_HOME}/lib64

      PATH=${CUDA_HOME}/bin:${PATH}

      export PATH

2. Create Virtualenv:

   virtualenv --system-site-packages ~/tensorflow

3. Activate Virtualenv:

   source ~/tensorflow/bin/activate

   # Prompt should change to start with "(tensorflow)$"

4. Install TensorFlow into Virtualenv:

   # Ubuntu/Linux 64-bit, GPU enabled, Python 2.7

   # Requires CUDA toolkit 7.5 and CuDNN v5. For other versions, see "Install from sources" below.

   (tensorflow)$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.10.0-cp27-none-linux_x86_64.whl

   (tensorflow)$ pip install --upgrade $TF_BINARY_URL

5. Test:

   (tensorflow)$ python

   >>> import tensorflow as tf

See Also

• Official site.
• Machine Learning
• Python