How to verify CuDNN installation?

前端 未结 9 961
余生分开走
余生分开走 2020-11-29 15:02

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 installe

9条回答
  •  没有蜡笔的小新
    2020-11-29 15:34

    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.

    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
    

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

提交回复
热议问题