一、安装显卡驱动

• 方式一：图形界面安装

等待安装即可

• 方式二：命令行安装

sudo apt-get install nvidia-driver-515

有些显卡只支持455

安装完成记得重启一下，然后验证一下：

nvidia-smi

若显示下图则说明安装成功~ 

 

 二、安装CUDA

• 下载地址：

CUDA Toolkit 11.7 Downloads | NVIDIA Developer

选择Linux->x86_64->Ubuntu->20.04->runfile(local)

wget https://developer.download.nvidia.com/compute/cuda/11.7.0/local_installers/cuda_11.7.0_515.43.04_linux.run

 

等待30m即可（取决于网速） 

• 安装cuda：

sudo sh cuda_11.7.0_515.43.04_linux.run

 选择Continue（上下移动选择，Enter确定）

 输入accept

 

在对应的选项上空格，然后移动到Install

显示如下就表示安装成功了！

• 添加环境变量

打开bashrc文件

sudo gedit ~/.bashrc

在文件的最后面添加下列路径 

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

 保存并关闭，生效环境变量

source ~/.bashrc

• 查看CUDA是否安装成功

nvcc -V

显示下列就说明安装成功！

 若没有安装成功，则显示如下：