получить версию CUDA и CUDNN для Windows с установкой Anaconda

Question

Существует версия tenorflow-gpu, установленная в Windows с использованием Anaconda, как проверить версию CUDA и CUDNN?Спасибо.

Answer 1

Вы также можете запустить conda list из командной строки anaconda:

conda list cudnn

# packages in environment at C:\Anaconda2:
#
# Name                    Version                   Build  Channel
cudnn                     6.0                           0

Answer 2

Хотя это не публично документированный API, вы можете в настоящее время получить к нему доступ следующим образом:

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