Step 1: Upgrade your Ubuntu

sudo apt update
sudo apt upgrade 

Step 2: List the recommended NVIDIA drivers

sudo apt install ubuntu-drivers-common
sudo ubuntu-drivers devices
# 输出
# driver   : nvidia-driver-535 - distro non-free recommended

Step 3: Install the driver nvidia-driver-535

sudo apt install nvidia-driver-535

Step 4: Reboot your system

sudo reboot now

Step 5: Check the driver installation

nvidia-smi

Step 6: Install GCC

sudo apt install gccgcc -v

Step 7: Install CUDA toolkit Ubuntu

访问下载安装：https://developer.nvidia.com/cuda-downloads

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.1-1_all.deb
sudo dpkg -i cuda-keyring_1.1-1_all.deb
sudo apt-get update
sudo apt-get -y install cuda

Step 8: Reboot your system

sudo reboot now

Step 9: Environment setup

nano ~/.bashrcexport PATH=/usr/local/cuda/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-12.2/lib64\${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}. ~/.bashrc

Step 10: Test the CUDA toolkit

nvcc -V

本文参考https://www.cherryservers.com/blog/install-cuda-ubuntu