Installing TensorFlow: Difference between revisions

From CCN Wiki
Jump to navigation Jump to search
No edit summary
No edit summary
Line 1: Line 1:
#First  [http://docs.nvidia.com/cuda/cuda-installation-guide-linux/#axzz4VZnqTJ2A install CUDA]
#First  [http://docs.nvidia.com/cuda/cuda-installation-guide-linux/#axzz4VZnqTJ2A install CUDA]
#*[https://devtalk.nvidia.com/default/topic/1026198/cuda-9-0-importerror-libcublas-so-8-0/ Apparently CUDA 9.0 won't fly]. Be sure to use CUDA 8.0 until 9.0 is validated. Sigh.
#Modify .bashrc to include CUDA in your PATH and LD_LIBRARY_PATH
#Modify .bashrc to include CUDA in your PATH and LD_LIBRARY_PATH
#*<code>export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64:$LD_LIBRARY_PATH</code>
#*<code>export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64:$LD_LIBRARY_PATH</code>

Revision as of 22:54, 15 November 2017

  1. First install CUDA
  2. Modify .bashrc to include CUDA in your PATH and LD_LIBRARY_PATH
    • export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64:$LD_LIBRARY_PATH
    • export PATH=/usr/local/cuda-9.0/bin:$PATH
    • export CUDA_HOME=/usr/local/cuda-9.0
  3. Install cuDNN
  4. Now install TensorFlow