instruction for the installation of the Nvidia driver + cuda + cudnn
nvidia driver 375, cuda 8.0, cudnn v5.1 [1]
- nvidia-375 driver:
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
sudo apt-get install nvidia-375
- cuda download web: https://developer.nvidia.com/cuda-toolkit-archive
- .deb:
sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb # or *-deb file
sudo apt-key add /var/cuda-repo-<version/Table>/7fa2af80.pub
sudo apt-get update
sudo apt-get install cuda
- or other format, such as .run
- .deb:
- cudnn download web: https://developer.nvidia.com/cudnn You need an account before download.
tar xvf cudnn*.tgz
cd cuda
sudo cp */*.h /usr/local/cuda/include/
sudo cp */libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
cuDNN without sudo (in your home folder) [2]
- set the environment variables LD_LIBRARY_PATH, LIBRARY_PATH and CPATH to the directory extracted from the download.
- If needed, separate multiple directories with : as in the PATH environment variable.
export CUDNN_ROOT=/home/YourUserName/libs/cudnn
export LD_LIBRARY_PATH=$CUDNN_ROOT/lib64:$LD_LIBRARY_PATH
export CPATH=$CUDNN_ROOT/include:$CPATH
export LIBRARY_PATH=$CUDNN_ROOT/lib64:$LD_LIBRARY_PATH
- cudnn
- if the cudnn was copied to CUDA_ROOT:
rm /usr/local/cuda/include/cudnn.h
rm /usr/local/cuda/lib64/libcudnn*
- if the cudnn was copied to CUDA_ROOT:
- cuda
/usr/local/cuda/bin/uninstallxxx
- or
sudo apt remove --purge cuda
if cuda was install using .deb file.
- nvidia driver:
sudo apt remove --purge nvidia*
- raise RuntimeError('Could not find cudnn library (looked for v5[.1])')
- solution: export LIBRARY_PATH=/usr/local/cuda/lib64:$LIBRARY_PATH