Difference between revisions of "Tensorflow"

From GISAXS
Jump to: navigation, search
(Virtualenv method)
(Virtualenv method)
Line 18: Line 18:
 
## Download CUDA runfile from [https://developer.nvidia.com/cuda-downloads the nVida site] (CUDA toolkit 7.5 runfile):
 
## Download CUDA runfile from [https://developer.nvidia.com/cuda-downloads 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
 
##: wget http://developer.download.nvidia.com/compute/cuda/7.5/Prod/local_installers/cuda_7.5.18_linux.run
 +
## Disable nouveau:
 +
##: sudo nano /etc/modprobe.d/blacklist-nouveau.conf
 +
##:: blacklist nouveau
 +
##:: options nouveau modeset=0
 
# Set up environment
 
# Set up environment
 
##: nano ~/.bashrc
 
##: nano ~/.bashrc
##: # Add these lines to the end of your .bashrc file:
+
##:: # Add these lines to the end of your .bashrc file:
##: export CUDA_HOME=/usr/local/cuda-7.5
+
##:: export CUDA_HOME=/usr/local/cuda-7.5
##: export LD_LIBRARY_PATH=${CUDA_HOME}/lib64
+
##:: export LD_LIBRARY_PATH=${CUDA_HOME}/lib64
##: PATH=${CUDA_HOME}/bin:${PATH}
+
##:: PATH=${CUDA_HOME}/bin:${PATH}
##: export PATH
+
##:: export PATH
 
# Create Virtualenv:
 
# Create Virtualenv:
##: virtualenv --system-site-packages ~/tensorflow
+
#: virtualenv --system-site-packages ~/tensorflow
 
# Activate Virtualenv:
 
# Activate Virtualenv:
##: source ~/tensorflow/bin/activate
+
#: source ~/tensorflow/bin/activate
 
##: # Prompt should change to start with "(tensorflow)$"
 
##: # Prompt should change to start with "(tensorflow)$"
 
# Install TensorFlow into Virtualenv:
 
# Install TensorFlow into Virtualenv:

Revision as of 15:08, 13 September 2016

TensorFlow is an open source library for deep learning.

Installation instructions

Virtualenv method

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
  3. 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
  4. Create Virtualenv:
    virtualenv --system-site-packages ~/tensorflow
  5. Activate Virtualenv:
    source ~/tensorflow/bin/activate
    1. # Prompt should change to start with "(tensorflow)$"
  6. 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
  7. Test:
    (tensorflow)$ python
    >>> import tensorflow as tf

See Also