NVIDIA

nvidia-tool kit -> for istallation of drivers

nvcc -> code compiler

cuda

cudann-> for deep learning

Install the nvidia-tool kit and nvidia compiler.
To check whether the nvidia compiler installed or not nvcc — version
check whether the nvidia-smi is working.
Then we have to install the cuda first.
1. https://developer.nvidia.com/cuda-downloads
To check the support of cuda , nvidia-driver and cudann versions.
1.https://docs.nvidia.com/deeplearning/cudnn/support-matrix/index.html
TO download the cudnn
Login into nvidia
https://developer.nvidia.com/rdp/cudnn-download
IF the above link is now working :-
1. https://developer.nvidia.com/downloads
2. Choose CUDA-X libraries
3. Choose Deep learning
4.CHoose NVIDIA cuDNN
5. Download CUDnn
Note : -
For pytorch and tensorflow cuda and cudann will come by default.

####################

locate cuda
which cuda
find cuda
###############