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

问题描述

我搜索了很多地方，但我得到的是如何安装它，而不是如何验证它是否已安装。我可以验证我的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!

PS。

这是一个caffe实现。目前一切正常工作没有启用CuDNN。

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

推荐答案

安装CuDNN只是将文件放入CUDA目录。

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

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屋！