commit 02f17fe8b7d1c58a3a2f224fe7ba6649c9ad84fd
author A. Unique TensorFlower <gardener@tensorflow.org> Sat Jul 07 06:59:19 2018 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> Sun Jul 08 17:14:19 2018 -0700
tree eb15a52cecc209139b3a50cf1beff59a7f5fd01b
parent 35287be3bb7daa0448af064f5d005a25201d6853

Set correct environment variable for CUDA install path

configure.py respects CUDA_TOOLKIT_PATH instead of CUDA_INSTALL_PATH

PiperOrigin-RevId: 203591214

configure.py

diff --git a/configure.py b/configure.py
index 31a83b4..03c03aa 100644
--- a/configure.py
+++ b/configure.py
@@ -835,6 +835,8 @@
                      '[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'