您可以按照以下步驟測試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上的安裝是否成功。如果遇到任何問題,請隨時告訴我。