您可以按照以下步骤测试CUDNN在Ubuntu上的安装是否成功:
- 首先,在终端中输入以下命令验证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)
- 接着,您可以尝试运行一个简单的CUDNN示例代码来测试CUDNN的功能。您可以使用以下命令克隆CUDNN示例代码的GitHub仓库:
git clone https://github.com/dusty-nv/jetson-inference
- 进入示例代码的目录并编译测试程序:
cd jetson-inference mkdir build cd build cmake .. make
- 编译完成后,您可以尝试运行一个CUDNN示例程序来验证CUDNN的功能:
./imagenet-console dog.jpg
以上步骤应该可以帮助您测试CUDNN在Ubuntu上的安装是否成功。如果遇到任何问题,请随时告诉我。