From 24cbb2a8ce16da1157e6ac64a7d6391bead6b633 Mon Sep 17 00:00:00 2001
From: "A. Unique TensorFlower" <gardener@tensorflow.org>
Date: Fri, 8 Sep 2017 07:45:44 -0700
Subject: [PATCH] Allow use of configure for gpu remote execution (i.e., no
 checks on local cuda)

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

diff --git a/configure.py b/configure.py
index ef5051d275e..7e5a84ecafc 100644
--- a/configure.py
+++ b/configure.py
@@ -995,7 +995,8 @@ def main():
     set_computecpp_toolkit_path(environ_cp)
 
   set_action_env_var(environ_cp, 'TF_NEED_CUDA', 'CUDA', False)
-  if environ_cp.get('TF_NEED_CUDA') == '1':
+  if (environ_cp.get('TF_NEED_CUDA') == '1' and
+      'TF_CUDA_CONFIG_REPO' not in environ_cp):
     set_tf_cuda_version(environ_cp)
     set_tf_cunn_version(environ_cp)
     set_tf_cuda_compute_capabilities(environ_cp)