Installing TensorFlow

From CCN Wiki
Revision as of 10:47, 16 November 2017 by 128.205.172.160 (talk)
Jump to navigation Jump to search
  1. First install CUDA
    • Apparently CUDA 9.0 won't fly. Be sure to use CUDA 8.0 until 9.0 is validated. Sigh.
    • CUDA 8.0 has been saved to the UBFS/Software folder. Install it using sudo apt-get install cuda-8-0. The logic behind installing this way is apparent when you look at Table 4 in the current iteration of the installation guide for CUDA 9.0
  1. Modify .bashrc to include CUDA_HOME and LD_LIBRARY_PATH
    • export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
    • export CUDA_HOME=/usr/local/cuda
  2. Modify .profile to include cuda/bin in your PATH
    • export PATH=/usr/local/cuda/bin:$PATH
  3. Install cuDNN
  4. Now install TensorFlow
Retrieved from "https://ccn-wiki.caset.buffalo.edu/index.php?title=Installing_TensorFlow&oldid=1246"