Ubuntu安装CUDA失败：驱动版本不匹配怎么办？

Question

在Ubuntu系统中安装CUDA时，如果遇到驱动版本不匹配的问题，通常是因为NVIDIA显卡驱动与CUDA Toolkit的版本不兼容。解决方法如下：首先，检查当前已安装的NVIDIA驱动版本，可通过命令`nvidia-smi`查看。接着，访问NVIDIA官网查询CUDA Toolkit版本所需的驱动版本。若驱动版本过低，则需要卸载旧驱动，使用`sudo apt-get remove --purge nvidia*`命令清理系统残留文件，然后添加NVIDIA官方PPA源，安装适配的驱动版本。例如，`sudo apt-get install nvidia-driver-xxx`（xxx为具体驱动版本号）。安装完成后重启系统，并重新安装对应版本的CUDA Toolkit。注意下载.run文件或.deb文件时要选择与驱动匹配的版本。最后验证安装是否成功，运行`nvcc --version`和`cuda-sample`示例程序进行测试。

Answer 1

1. 问题概述

在Ubuntu系统中安装CUDA时，驱动版本不匹配是一个常见的问题。这通常是因为NVIDIA显卡驱动与CUDA Toolkit的版本不兼容。为了确保CUDA能够正常运行，必须保证驱动和CUDA Toolkit的版本匹配。

• 常见问题：运行`nvidia-smi`命令时显示错误信息，或者CUDA相关程序无法启动。
• 影响范围：任何使用NVIDIA GPU进行深度学习或高性能计算的开发者。

2. 检查当前环境

首先需要确认当前系统的NVIDIA驱动版本是否满足CUDA Toolkit的要求。

# 查看当前已安装的NVIDIA驱动版本
nvidia-smi
    

上述命令会输出类似以下内容：

Driver Version	CUDA Version
470.82	11.4

如果驱动版本过低，可能需要升级到适配的版本。

3. 查询兼容性

访问NVIDIA官网，查询CUDA Toolkit版本所需的驱动版本。例如，CUDA 11.8要求驱动版本至少为520。

1. 进入CUDA Toolkit Archive页面。
2. 选择目标CUDA版本，查看其系统需求。

若发现当前驱动版本不符合要求，则需要卸载旧驱动并安装新驱动。

4. 卸载旧驱动

使用以下命令清理系统中残留的NVIDIA驱动文件：

sudo apt-get remove --purge nvidia*
sudo apt-get autoremove
    

这一步非常重要，避免旧驱动文件干扰新驱动的安装。

5. 安装适配驱动

添加NVIDIA官方PPA源，并安装适配的驱动版本：

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
sudo apt-get install nvidia-driver-xxx
    

其中，`xxx`是具体驱动版本号，例如525。

安装完成后重启系统：

sudo reboot
    

6. 安装CUDA Toolkit

根据驱动版本选择对应的CUDA Toolkit版本。可以通过.run或.deb文件安装：

以下载.deb文件为例：

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda-repo-ubuntu2004-11-8-local_11.8.0-520.61-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2004-11-8-local_11.8.0-520.61-1_amd64.deb
sudo apt-get update
sudo apt-get -y install cuda
    

确保环境变量配置正确：

echo 'export PATH=/usr/local/cuda-11.8/bin:$PATH' >> ~/.bashrc
echo 'export LD_LIBRARY_PATH=/usr/local/cuda-11.8/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc
source ~/.bashrc
    

7. 验证安装

最后验证CUDA是否安装成功：

nvcc --version
    

运行示例程序测试：

cd /usr/local/cuda-11.8/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery
    

如果输出GPU设备信息且无错误提示，则表示安装成功。

8. 流程图

graph TD;
            A[检查驱动版本] --> B{版本是否匹配};
            B -->|是| C[安装CUDA];
            B -->|否| D[卸载旧驱动];
            D --> E[安装适配驱动];
            E --> C;
            C --> F[验证安装];