QBoard » Artificial Intelligence & ML » AI and ML - PyTorch » How to verify CuDNN installation?
User Dashboard

How to verify CuDNN installation?

Back To Topics
Tags : None
  • 126
    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.
    This is for a caffe implementation. Currently everything is working without CuDNN enabled.
      November 3, 2021 5:28 PM IST
    0
  • 317

    How about checking with python code:

    from tensorflow.python.platform import build_info as tf_build_info

print(tf_build_info.cudnn_version_number)
# 7 in v1.10.0
      December 23, 2021 2:08 PM IST
    0
  • 461
    I have cuDNN 8.0 and none of the suggestions above worked for me. The desired information was in /usr/include/cudnn_version.h, so

    cat /usr/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
​



    did the trick.

     
      December 7, 2021 12:32 PM IST
    0
  • 319 1

    On Ubuntu 20.04LTS:

    cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR

     

    returned the expected results

     
      November 11, 2021 2:32 PM IST
    0