/driver_cuda_cudnn

instruction for the installation of the Nvidia driver + cuda + cudnn

driver_cuda_cudnn

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
  • 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

uninstall:

  • cudnn
    • if the cudnn was copied to CUDA_ROOT:
      • rm /usr/local/cuda/include/cudnn.h
      • rm /usr/local/cuda/lib64/libcudnn*
  • 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*

errors:

  • raise RuntimeError('Could not find cudnn library (looked for v5[.1])')
    • solution: export LIBRARY_PATH=/usr/local/cuda/lib64:$LIBRARY_PATH