Allow use of configure for gpu remote execution (i.e., no checks on local cuda)
PiperOrigin-RevId: 167995887
diff --git a/configure.py b/configure.py
index ef5051d..7e5a84e 100644
--- a/configure.py
+++ b/configure.py
@@ -995,7 +995,8 @@
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)