Installing TensorFlow: Difference between revisions
Jump to navigation
Jump to search
No edit summary |
No edit summary |
||
| Line 7: | Line 7: | ||
#Modify .profile to include cuda/bin in your PATH | #Modify .profile to include cuda/bin in your PATH | ||
#*<code>export PATH=/usr/local/cuda/bin:$PATH</code> | #*<code>export PATH=/usr/local/cuda/bin:$PATH</code> | ||
#Install [https://developer.nvidia.com/rdp/cudnn-download cuDNN] | #Install [https://developer.nvidia.com/rdp/cudnn-download cuDNN 6.0 (NOT 7!)] | ||
#Now [https://www.tensorflow.org/install/install_linux install TensorFlow ] | #Now [https://www.tensorflow.org/install/install_linux install TensorFlow ] | ||
Latest revision as of 11:12, 16 November 2017
- 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
- Modify .bashrc to include CUDA_HOME and LD_LIBRARY_PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATHexport CUDA_HOME=/usr/local/cuda
- Modify .profile to include cuda/bin in your PATH
export PATH=/usr/local/cuda/bin:$PATH
- Install cuDNN 6.0 (NOT 7!)
- Now install TensorFlow