get the CUDA and CUDNN version on windows with Anaconda installe

后端 未结 3 1522
遇见更好的自我
遇见更好的自我 2020-12-16 05:44

There is a tensorflow-gpu version installed on Windows using Anaconda, how to check the CUDA and CUDNN version of it? Thanks.

3条回答
  •  时光取名叫无心
    2020-12-16 06:35

    Although not a public documented API, you can currently access it like this:

    from tensorflow.python.platform import build_info as tf_build_info
    print(tf_build_info.cuda_version_number)
    # 9.0 in v1.10.0
    print(tf_build_info.cudnn_version_number)
    # 7 in v1.10.0
    

提交回复
热议问题