您可以按照以下步骤测试CUDNN在Ubuntu上的安装是否成功:
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)
git clone https://github.com/dusty-nv/jetson-inference
cd jetson-inference
mkdir build
cd build
cmake ..
make
./imagenet-console dog.jpg
以上步骤应该可以帮助您测试CUDNN在Ubuntu上的安装是否成功。如果遇到任何问题,请随时告诉我。