Fix reported cuDNN default version during configuration.
PiperOrigin-RevId: 215272308
This commit is contained in:
parent
c86f594135
commit
44acd839c5
@ -884,7 +884,7 @@ def set_tf_cudnn_version(environ_cp):
|
|||||||
"""Set CUDNN_INSTALL_PATH and TF_CUDNN_VERSION."""
|
"""Set CUDNN_INSTALL_PATH and TF_CUDNN_VERSION."""
|
||||||
ask_cudnn_version = (
|
ask_cudnn_version = (
|
||||||
'Please specify the cuDNN version you want to use. '
|
'Please specify the cuDNN version you want to use. '
|
||||||
'[Leave empty to default to cuDNN %s.0]: ') % _DEFAULT_CUDNN_VERSION
|
'[Leave empty to default to cuDNN %s]: ') % _DEFAULT_CUDNN_VERSION
|
||||||
|
|
||||||
for _ in range(_DEFAULT_PROMPT_ASK_ATTEMPTS):
|
for _ in range(_DEFAULT_PROMPT_ASK_ATTEMPTS):
|
||||||
tf_cudnn_version = get_from_env_or_user_or_default(
|
tf_cudnn_version = get_from_env_or_user_or_default(
|
||||||
|
Loading…
Reference in New Issue
Block a user