diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-07-07 06:59:19 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-08 17:14:19 -0700 |
commit | 02f17fe8b7d1c58a3a2f224fe7ba6649c9ad84fd (patch) | |
tree | eb15a52cecc209139b3a50cf1beff59a7f5fd01b /configure.py | |
parent | 35287be3bb7daa0448af064f5d005a25201d6853 (diff) |
Set correct environment variable for CUDA install path
configure.py respects CUDA_TOOLKIT_PATH instead of CUDA_INSTALL_PATH
PiperOrigin-RevId: 203591214
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' |