Installing TensorFlow: Difference between revisions
Jump to navigation
Jump to search
No edit summary |
No edit summary |
||
(One intermediate revision by the same user not shown) | |||
Line 2: | Line 2: | ||
#*[https://devtalk.nvidia.com/default/topic/1026198/cuda-9-0-importerror-libcublas-so-8-0/ Apparently CUDA 9.0 won't fly]. Be sure to use CUDA 8.0 until 9.0 is validated. Sigh. | #*[https://devtalk.nvidia.com/default/topic/1026198/cuda-9-0-importerror-libcublas-so-8-0/ 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 <code>sudo apt-get install cuda-8-0</code>. The logic behind installing this way is apparent when you look at [http://docs.nvidia.com/cuda/cuda-installation-guide-linux/#package-manager-metas Table 4] in the current iteration of the installation guide for CUDA 9.0 | #*CUDA 8.0 has been saved to the UBFS/Software folder. Install it using <code>sudo apt-get install cuda-8-0</code>. The logic behind installing this way is apparent when you look at [http://docs.nvidia.com/cuda/cuda-installation-guide-linux/#package-manager-metas Table 4] in the current iteration of the installation guide for CUDA 9.0 | ||
#Modify .bashrc to include CUDA_HOME and LD_LIBRARY_PATH | #Modify .bashrc to include CUDA_HOME and LD_LIBRARY_PATH | ||
#*<code>export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH</code> | #*<code>export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH</code> | ||
Line 8: | 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_PATH
export 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