Ubuntu环境下搭建CUDA编程环境

一、安装NVIDIA驱动

1.查看本机显卡型号

lspci | grep -i nvidia

显示： NVIDIA Corporation GP102GL [Tesla P40] (rev a1)

2. 查看显卡支持的驱动

检测你的NVIDIA图形卡和推荐的驱动程序的模型。执行命令：

# 下载驱动管理工具
apt install ubuntu-drivers-common
# 查看显卡支持的驱动
ubuntu-drivers devices 

3.安装驱动

你可以选择，安装所有推荐的驱动，如下命令

 sudo ubuntu-drivers autoinstall

也可以直接安装最新的

sudo apt install nvidia-driver-550

4.查看驱动信息

运行nvidia-smi查看驱动信息

nvidia-smi

二、安装CUDA

1.选择CUDA版本

进入cuda官网，根据自己的设备选择对应的选项
https://developer.nvidia.com/cuda-toolkit-archive

不知道自己显卡对应什么版本的执行命令nvidia-smi。我这里为12.8版本

然后找到12.8版本的CUDA

2.下载并安装CUDA

根据自己的系统进行选择，我这里是ubuntu22

根据提示选择下载方式，并输入对应的指令，如上图。

wget https://developer.download.nvidia.com/compute/cuda/12.8.0/local_installers/cuda_12.8.0_570.86.10_linux.run
sudo sh cuda_12.8.0_570.86.10_linux.run

若提示

Existing package manager installation of the driver found. It is strongly recommended that you remove this before continuing
则选择continue，并去除driver项，之后选择install

有如下输出则表示成功

3.修改~/.bashrc

sudo vim ~/.bashrc
source ~/.bashrc

按照图中输出Summary后面内容的提示，在~/.bashrc最后添加：

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

4.验证是否安装成功

输入：

nvcc --version

则安装成功！

5.如果需要卸载

cd usr/local/cuda-12.8/bin    //打开cuda-12.4安装位置
sudo ./cuda-uninstaller       //运行卸载程序
vim ~/.bashrc //删除前面添加的东西