Installing TensorFlow: Difference between revisions
Jump to navigation
Jump to search
No edit summary |
No edit summary |
||
| Line 5: | Line 5: | ||
#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> | ||
#*<code>export PATH=/usr/local/cuda | #*<code>export PATH=/usr/local/cuda/bin:$PATH</code> | ||
#*<code>export CUDA_HOME=/usr/local/cuda | #*<code>export CUDA_HOME=/usr/local/cuda</code> | ||
#Install [https://developer.nvidia.com/rdp/cudnn-download cuDNN] | #Install [https://developer.nvidia.com/rdp/cudnn-download cuDNN] | ||
#Now [https://www.tensorflow.org/install/install_linux install TensorFlow ] | #Now [https://www.tensorflow.org/install/install_linux install TensorFlow ] | ||
Revision as of 10:37, 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 in your PATH and LD_LIBRARY_PATH
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64:$LD_LIBRARY_PATHexport PATH=/usr/local/cuda/bin:$PATHexport CUDA_HOME=/usr/local/cuda
- Install cuDNN
- Now install TensorFlow