Installing TensorFlow

From CCN Wiki
Revision as of 10:36, 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 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 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
  2. Install cuDNN
  3. Now install TensorFlow
Retrieved from "https://ccn-wiki.caset.buffalo.edu/index.php?title=Installing_TensorFlow&oldid=1241"