-
Install ubuntu drivers:
-
Install cuda
- Download the cuda toolkit from the nvidia website
- Select the version
- Run the installation instructions
$ wget https://developer.download.nvidia.com/compute/cuda/11.6.2/local_installers/cuda_11.6.2_510.47.03_linux.run
- Install cuda
$ sudo sh cuda_11.6.2_510.47.03_linux.run
. - Add PATH to ~/.bashrc following the instruction.
- Check if CUDA is installed
$ nvcc -V
-
Install cudnn
- Download cudnn from nvidia
- Extract the downloaded file and copy to the cuda directory.
- Check the cudnn version ```$ cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2