ubuntu

cudnn ubuntu安装后测试

小樊
174
2024-07-05 00:37:30
栏目: 智能运维

您可以按照以下步骤测试CUDNN在Ubuntu上的安装是否成功:

  1. 首先,在终端中输入以下命令验证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)
  1. 接着,您可以尝试运行一个简单的CUDNN示例代码来测试CUDNN的功能。您可以使用以下命令克隆CUDNN示例代码的GitHub仓库:
git clone https://github.com/dusty-nv/jetson-inference
  1. 进入示例代码的目录并编译测试程序:
cd jetson-inference
mkdir build
cd build
cmake ..
make
  1. 编译完成后,您可以尝试运行一个CUDNN示例程序来验证CUDNN的功能:
./imagenet-console dog.jpg

以上步骤应该可以帮助您测试CUDNN在Ubuntu上的安装是否成功。如果遇到任何问题,请随时告诉我。

0
看了该问题的人还看了