Set TF_CUDA_PATHS.
(environ_cp)
| 865 | |
| 866 | |
| 867 | def set_tf_cuda_paths(environ_cp): |
| 868 | """Set TF_CUDA_PATHS.""" |
| 869 | ask_cuda_paths = ( |
| 870 | 'Please specify the comma-separated list of base paths to look for CUDA ' |
| 871 | 'libraries and headers. [Leave empty to use the default]: ') |
| 872 | tf_cuda_paths = get_from_env_or_user_or_default(environ_cp, 'TF_CUDA_PATHS', |
| 873 | ask_cuda_paths, '') |
| 874 | if tf_cuda_paths: |
| 875 | environ_cp['TF_CUDA_PATHS'] = tf_cuda_paths |
| 876 | |
| 877 | |
| 878 | def set_tf_cuda_version(environ_cp): |
no test coverage detected