在ubuntu20中安装cuda

安装显卡驱动

1. 添加官方源并更新

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update

2. 查看推荐的驱动版本

ubuntu-drivers devices

3. 安装推荐的版本

sudo apt instal nvidia-driver-510

4. 使用nvidia-smi检查显卡信息

安装CUDA

1. 查看nvidia驱动对应的CUDA版本：https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/
   

2. 选择对应版本的cuda：https://developer.nvidia.com/cuda-toolkit-archive
   

3. 选择系统版本，根据提示进行下载
   

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/11.6.2/local_installers/cuda-repo-ubuntu2004-11-6-local_11.6.2-510.47.03-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2004-11-6-local_11.6.2-510.47.03-1_amd64.deb
sudo apt-key add /var/cuda-repo-ubuntu2004-11-6-local/7fa2af80.pub
sudo apt-get update
sudo apt-get -y install cuda

4. 打开~/.zshrc，添加下列语句，保存并退出后运行source ~/.zshrc

export PATH=/usr/local/cuda/bin:${PATH}
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:${LD_LIBRARY_PATH}

5. 运行nvcc --version，判断是否安装成功