Cuda installation

CUDA

You can download CUDA10.2 from here. You can follow the instruction.

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-ubuntu1804.pin

sudo mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600

sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub

sudo add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/ /"

sudo apt-get update

sudo apt-get -y install cuda

cuDNN

You can download specific version from here. For example, the tested version is with libcudnn8_8.0.0.180-1+cuda10.2_amd64.deb.

Then install them using the command below.

sudo dpkg -i libcudnn8_8.0.0.180-1+cuda10.2_amd64.deb

sudo dpkg -i libcudnn8-dev_8.0.0.180-1+cuda10.2_amd64.deb