一、安装nvidia驱动 

# 添加nvidia驱动ppa库
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update# 查找推荐版本
sudo ubuntu-drivers devices# 安装推荐版本
sudo apt install nvidia-driver-560# 检验nvidia驱动是否安装
nvidia-smi

二、安装cudatoolkit（cuda）
官网：https://developer.nvidia.com/cuda-toolkit-archive

• 查看nvidia驱动支持的cuda版本

• 按照官网提示去操作

• 配置环境变量 

# 如果shell是zsh
sudo vim ~/.zshrc# 添加如下内容：export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH# 刷新配置文件
source ~/.zshrc

 

• 验证是否安装成功

nvcc --version