Set TF_CUDNN_VERSION.
(environ_cp)
| 888 | |
| 889 | |
| 890 | def set_tf_cudnn_version(environ_cp): |
| 891 | """Set TF_CUDNN_VERSION.""" |
| 892 | ask_cudnn_version = ( |
| 893 | 'Please specify the cuDNN version you want to use. ' |
| 894 | '[Leave empty to default to cuDNN %s]: ') % _DEFAULT_CUDNN_VERSION |
| 895 | tf_cudnn_version = get_from_env_or_user_or_default(environ_cp, |
| 896 | 'TF_CUDNN_VERSION', |
| 897 | ask_cudnn_version, |
| 898 | _DEFAULT_CUDNN_VERSION) |
| 899 | environ_cp['TF_CUDNN_VERSION'] = tf_cudnn_version |
| 900 | |
| 901 | |
| 902 | def is_cuda_compatible(lib, cuda_ver, cudnn_ver): |
no test coverage detected