且构网

分享程序员开发的那些事...
且构网 - 分享程序员编程开发的那些事

如何验证CuDNN安装?

更新时间:2021-10-28 21:50:58

安装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.