您可以按照以下步骤测试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上的安装是否成功。如果遇到任何问题,请随时告诉我。
亿速云「云服务器」,即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘,价格低至29元/月。点击查看>>
推荐阅读:ubuntu中cudnn怎么使用