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

Question

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

1

В Windows используется версия tensorflow-gpu, использующая Anaconda, как проверить версию CUDA и CUDNN? Благодарю.

user297850 19 сен. 2018, в 18:37

Источник

0

Используйте список conda, он покажет вам версии пакета
talonmies 19 сен. 2018, в 16:42

Теги:

python

tensorflow

anaconda

gpu

cudnn

2 ответа

Ещё вопросы

Используйте список conda, он покажет вам версии пакета

jdehesa · Answer 1 · 2018-09-19T13-50-00.000Z

Хотя это не общедоступный документированный 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

Это не работает для меня (тензор потока 1.13)
@ iyop45 Спасибо за отзыв, вы правы, последние версии build_info больше не включают эту информацию в build_info . Я открыл вопрос об этом , это, кажется, является следствием отказа от сборки CMake для Windows в пользу Bazel.
@ iyop45 Это было только что исправлено в этом коммите. Не уверен, какой релиз первым покажет исправление.

John H · Answer 2 · 2018-12-21T16-39-00.000Z

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

conda list cudnn

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