查看是否正确安装了相应的 CUDA 和 cuDNN 驱动

要检查是否正确安装了 CUDA 和 cuDNN 驱动，可以按照以下步骤进行：

1. 检查 CUDA 是否已正确安装：

在命令行中输入以下命令：

nvcc --version

如果输出类似于 “nvcc: NVIDIA (R) Cuda compiler driver” 的信息，则说明 CUDA 已正确安装。

2. 检查 cuDNN 是否已正确安装：

在命令行中输入以下命令：

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

如果输出类似于以下信息，则说明 cuDNN 已正确安装：

#define CUDNN_MAJOR 7
#define CUDNN_MINOR 6
#define CUDNN_PATCHLEVEL 5
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)

如果您发现 CUDA 或 cuDNN 未正确安装，请参考 NVIDIA 官方文档以获取更多信息。

相关问题

在Ubuntu系统上如何正确安装NVIDIA显卡驱动、CUDA和CUDNN，并确保系统环境变量配置得当以支持GPU计算？

为了在Ubuntu系统上充分利用NVIDIA GPU进行深度学习和高性能计算，你需要按照以下步骤来安装NVIDIA显卡驱动、CUDA和CUDNN，并配置相应的环境变量。

参考资源链接：[Ubuntu系统详细安装教程：显卡驱动+CUDA+CUDNN+FTP](https://wenku.csdn.net/doc/6475b0f0543f844488ffd472?spm=1055.2569.3001.10343)

首先，使用`lspci -k | grep -A2 -i

参考资源链接：[Ubuntu系统详细安装教程：显卡驱动+CUDA+CUDNN+FTP](https://wenku.csdn.net/doc/6475b0f0543f844488ffd472?spm=1055.2569.3001.10343)