Install CUDA

# Uninstall Old Driver
sudo rm /etc/apt/sources.list.d/cuda*

sudo apt remove --autoremove nvidia-cuda-toolkit
sudo apt remove --autoremove nvidia-*

sudo rm -rf /usr/local/cuda*

sudo apt-get purge nvidia*

sudo apt-get update
sudo apt-get autoremove
sudo apt-get autoclean

# Show recomended driver
ubuntu-drivers devices

  1. install CUDA and Driver https://developer.nvidia.com/cuda-downloads

    Can choose a flavor!

    Untitled

    reboot!

  2. login and download cuDNN https://developer.nvidia.com/rdp/cudnn-download

# Check Driver
nvidia-smi

# check cuda
nvcc --version
# may add to PATH
export PATH=/usr/local/cuda-12.4/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-12.4/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

# check cudnn
ls /usr/local/cuda/include/cudnn.h
ls /usr/local/cuda/lib64/libcudnn*