Ubuntu安装英伟达驱动与CUDA

准备阶段

1. 准备好英伟达GPU驱动

   https://www.nvidia.cn/Download/index.aspx?lang=cn

2. 确认需要的驱动

安装驱动

1. 首先安装前，需要确认Ubuntu有安装gcc g++ make

sudo apt install gcc g++ make

2. 将驱动上传到Ubuntu
3. 给驱动添加可执行权限

chmod +x NVIDIA-Linux-*.run

4. 安装驱动

sudo ./NVIDIA-Linux-*.run

5. 安装完成后，reboot 重启Ubuntu，验证安装

nvidia-smi

安装CUDA

1. 访问英伟达CUDA官网https://developer.nvidia.com/cuda-toolkit-archive

2. 通过Nvidia-smi 确认CUDA对应版本

3. 选择对应的cuda版本

   

4. 使用官网提供的runfile方法

   wget https://developer.download.nvidia.com/compute/cuda/12.2.2/local_installers/cuda_12.2.2_535.104.05_linux.run      #下载run文件
   sudo sh cuda_12.2.2_535.104.05_linux.run      #运行run文件安装cuda

5. 编辑环境变量

sudo vim ~/.bashrc

在文件末尾添加

#/usr/local/cuda-12.2视 真实环境为准
export PATH=/usr/local/cuda-12.2/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-12.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
export CUDA_HOME=/usr/local/cuda-12.2

更新系统环境

source ~/.bashrc
sudo ldconfig

6. 验证安装

nvcc -V