diff options
author | 2017-09-08 07:45:44 -0700 | |
---|---|---|
committer | 2017-09-08 07:49:26 -0700 | |
commit | 24cbb2a8ce16da1157e6ac64a7d6391bead6b633 (patch) | |
tree | 7994d082d2f3c53f610f94cf97b91059d615844b /configure.py | |
parent | 10b966b95d55cad03c4f460619dfc8207989f6e4 (diff) |
Allow use of configure for gpu remote execution (i.e., no checks on local cuda)
PiperOrigin-RevId: 167995887
Diffstat (limited to 'configure.py')
-rw-r--r-- | configure.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/configure.py b/configure.py index ef5051d275..7e5a84ecaf 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) |