Installing TensorFlow

From CCN Wiki
Revision as of 22:54, 15 November 2017 by Chris (talk | contribs)
Jump to navigation Jump to search
  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