diff options
Diffstat (limited to 'configure.py')
-rw-r--r-- | configure.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/configure.py b/configure.py index 31a83b4a15..03c03aad9c 100644 --- a/configure.py +++ b/configure.py @@ -835,6 +835,8 @@ def set_tf_cuda_version(environ_cp): '[Default is %s]: ') % (tf_cuda_version, default_cuda_path) cuda_toolkit_path = get_from_env_or_user_or_default( environ_cp, 'CUDA_TOOLKIT_PATH', ask_cuda_path, default_cuda_path) + if is_windows() or is_cygwin(): + cuda_toolkit_path = cygpath(cuda_toolkit_path) if is_windows(): cuda_rt_lib_path = 'lib/x64/cudart.lib' |