In a terminal, type:
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update
sudo ubuntu-drivers autoinstall
reboot
sudo apt install nvidia-cuda-toolkit gcc-6
nvcc --version
checking :
nvcc --version
cudnn for cuda-8 :
https://developer.nvidia.com/compute/machine-learning/cudnn/secure/v7.1.3/prod/8.0_20180414/cudnn-8.0-linux-x64-v7.1
--------------------------------------------
tar xzvf cudnn-8.0-linux-x64-v7.1.tgz
sudo cp cuda/lib64/* /usr/lib/cuda/lib64/
sudo cp cuda/include/* /usr/lib/cuda/include/
sudo chmod a+r /usr/lib/cuda/lib64/libcudnn*
sudo chmod a+r /usr/lib/cuda/include/cudnn.h
Checking :
cat /usr/lib/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2