Linux下如何确认CUDA安装成功？命令是什么？

Question

在Linux下如何确认CUDA安装成功？命令是什么？ 在Linux系统中，完成CUDA安装后，需要验证其是否正确安装并能正常工作。首先，确认NVIDIA驱动程序已正确安装，通过运行`nvidia-smi`命令查看显卡状态与驱动信息。接着，验证CUDA Toolkit的安装，打开终端输入`nvcc --version`或`nvcc -V`，显示CUDA版本信息则表示安装成功。此外，可通过编译和运行CUDA示例程序进一步验证。进入CUDA样本目录（如`/usr/local/cuda-xx/samples`），使用`make`命令构建示例程序，运行`deviceQuery`样例，若输出“Result = PASS”，表明CUDA环境配置无误。最后，确保环境变量已正确设置，检查`.bashrc`文件中是否添加了CUDA路径。这些步骤有助于全面确认CUDA在Linux下的安装状态。

Answer 1

1. 初步验证：NVIDIA驱动程序安装

在Linux系统中，CUDA的运行依赖于NVIDIA驱动程序。要确认驱动程序是否正确安装，可以通过以下命令查看显卡状态和驱动信息：

nvidia-smi

如果命令返回显卡的状态信息、驱动版本以及CUDA兼容性信息，则表明NVIDIA驱动程序已成功安装。

2. 验证CUDA Toolkit安装

接下来，验证CUDA Toolkit是否正确安装。打开终端并输入以下命令以检查CUDA编译器版本：

nvcc --version

或

nvcc -V

如果命令返回CUDA版本号（例如“release 11.8, V11.8.89”），则表示CUDA Toolkit已成功安装。

3. 进一步验证：编译和运行示例程序

为了确保CUDA环境配置无误，可以编译和运行示例程序。以下是具体步骤：

1. 进入CUDA样本目录（通常位于/usr/local/cuda-xx/samples）。
2. 使用make命令构建示例程序：

cd /usr/local/cuda-xx/samples/1_Utilities/deviceQuery
make

构建完成后，运行deviceQuery样例：

./deviceQuery

如果输出包含“Result = PASS”，则表明CUDA环境配置正确。

4. 环境变量配置检查

最后，确保环境变量已正确设置。检查.bashrc文件中是否添加了CUDA路径：

export PATH=/usr/local/cuda-xx/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-xx/lib64:$LD_LIBRARY_PATH

完成修改后，执行以下命令使更改生效：

source ~/.bashrc

5. 综合分析与流程图

为便于理解整个验证过程，以下是一个流程图描述：

graph TD;
            A[开始] --> B{是否安装NVIDIA驱动？};
            B --是--> C[运行nvidia-smi];
            B --否--> E[重新安装驱动];
            C --> D{是否显示驱动信息？};
            D --是--> F[检查CUDA Toolkit];
            D --否--> G[排查驱动问题];
            F --> H[运行nvcc --version];
            H --> I{是否显示版本号？};
            I --是--> J[编译并运行示例程序];
            I --否--> K[重新安装CUDA];
            J --> L{示例程序是否通过？};
            L --是--> M[验证成功];
            L --否--> N[检查环境变量];

通过以上步骤，您可以全面确认CUDA在Linux下的安装状态。