Set TF_CUDA_VERSION.
(environ_cp)
| 876 | |
| 877 | |
| 878 | def set_tf_cuda_version(environ_cp): |
| 879 | """Set TF_CUDA_VERSION.""" |
| 880 | ask_cuda_version = ( |
| 881 | 'Please specify the CUDA SDK version you want to use. ' |
| 882 | '[Leave empty to default to CUDA %s]: ') % _DEFAULT_CUDA_VERSION |
| 883 | tf_cuda_version = get_from_env_or_user_or_default(environ_cp, |
| 884 | 'TF_CUDA_VERSION', |
| 885 | ask_cuda_version, |
| 886 | _DEFAULT_CUDA_VERSION) |
| 887 | environ_cp['TF_CUDA_VERSION'] = tf_cuda_version |
| 888 | |
| 889 | |
| 890 | def set_tf_cudnn_version(environ_cp): |
no test coverage detected