cudnn ubuntu安装后测试

   2024-10-07 7250
核心提示:您可以按照以下步骤测试CUDNN在Ubuntu上的安装是否成功:首先,在终端中输入以下命令验证CUDNN是否正确安装:cat /usr/local/cud

您可以按照以下步骤测试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-inferencemkdir buildcd buildcmake ..make
编译完成后,您可以尝试运行一个CUDNN示例程序来验证CUDNN的功能:
./imagenet-console dog.jpg

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

 
举报打赏
 
更多>同类物流大全
推荐图文
推荐物流大全
点击排行

网站首页  |  关于我们  |  联系方式网站留言    |  赣ICP备2021007278号