From 02f17fe8b7d1c58a3a2f224fe7ba6649c9ad84fd Mon Sep 17 00:00:00 2001
From: "A. Unique TensorFlower" <gardener@tensorflow.org>
Date: Sat, 7 Jul 2018 06:59:19 -0700
Subject: Set correct environment variable for CUDA install path

configure.py respects CUDA_TOOLKIT_PATH instead of CUDA_INSTALL_PATH

PiperOrigin-RevId: 203591214
---
 configure.py | 2 ++
 1 file changed, 2 insertions(+)

(limited to 'configure.py')

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'
-- 
cgit v1.2.3