如何验证 CuDNN 安装? [英] How to verify CuDNN installation?

查看:47
本文介绍了如何验证 CuDNN 安装?的处理方法,对大家解决问题具有一定的参考价值,需要的朋友们下面随着小编来一起学习吧!

问题描述

我已经搜索了很多地方,但我得到的只是如何安装它,而不是如何验证它是否已安装.我可以验证我的 NVIDIA 驱动程序已安装,并且 CUDA 已安装,但我不知道如何验证 CuDNN 已安装.非常感谢您的帮助,谢谢!

I have searched many places but ALL I get is HOW to install it, not how to verify that it is installed. I can verify my NVIDIA driver is installed, and that CUDA is installed, but I don't know how to verify CuDNN is installed. Help will be much appreciated, thanks!

附言.
这是针对 caffe 实现的.目前,在没有启用 CuDNN 的情况下一切正常.

PS.
This is for a caffe implementation. Currently everything is working without CuDNN enabled.

推荐答案

安装 CuDNN 只需要将文件放在 CUDA 目录中.如果您在安装 caffe 时正确指定了路由和 CuDNN 选项,它将与 CuDNN 一起编译.

Installing CuDNN just involves placing the files in the CUDA directory. If you have specified the routes and the CuDNN option correctly while installing caffe it will be compiled with CuDNN.

您可以使用 cmake 进行检查.创建一个目录 caffe/build 并从那里运行 cmake ...如果配置正确,您将看到以下几行:

You can check that using cmake. Create a directory caffe/build and run cmake .. from there. If the configuration is correct you will see these lines:

-- Found cuDNN (include: /usr/local/cuda-7.0/include, library: /usr/local/cuda-7.0/lib64/libcudnn.so)

-- NVIDIA CUDA:
--   Target GPU(s)     :   Auto
--   GPU arch(s)       :   sm_30
--   cuDNN             :   Yes

如果一切正确,只需运行 make 命令从那里安装 caffe.

If everything is correct just run the make orders to install caffe from there.

这篇关于如何验证 CuDNN 安装?的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持IT屋!

查看全文
登录 关闭
扫码关注1秒登录
发送“验证码”获取 | 15天全站免登陆