Ubuntu安装GPU驱动、CUDA

 一、GPU驱动安装

 1、根据要安装cuda版本确定GPU版本

官方对应文档：https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html

 比如我安装的cuda11.8,就需要安装>=520.61.05

2、安装

进入软件与更新安装， 软件与更新——附加驱动，选择要安装的驱动，点击应用更改。

3、重启

安装好后要重启，重启后终端输入nvidia-smi就能查看驱动信息。

二、安装cuda

1、下载cuda版本

CUDA下载：https://developer.nvidia.com/cuda-toolkit-archive

选择要安装的版本，根据下面的命令安装

2、添加环境变量

 打开./bashrc

sudo gedit ~/.bashrc

注意将下面cuda版本替换成你的

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

3、验证

nvcc -V