温馨提示×

如何验证ubuntu中cudnn安装

小樊
1153
2024-07-05 00:32:19
栏目: 智能运维

要验证Ubuntu中是否正确安装了cuDNN,可以按照以下步骤进行验证:

  1. 打开终端窗口,并输入以下命令来查看cuDNN的版本信息:
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
  1. 如果cuDNN已经正确安装,将会显示cuDNN的版本信息,例如:
#define CUDNN_MAJOR 8
#define CUDNN_MINOR 0
#define CUDNN_PATCHLEVEL 5
  1. 可以使用以下命令来测试cuDNN是否能够正常运行:
cd /usr/src/cudnn_samples_v8/mnistCUDNN
sudo make
./mnistCUDNN
  1. 如果cuDNN能够正常运行并输出结果,那么说明cuDNN已经正确安装在Ubuntu系统中。

通过以上步骤可以验证在Ubuntu中是否正确安装了cuDNN。

0