OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2026-01-15 12:15:41 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Allow use of configure for gpu remote execution (i.e., no checks on local cuda)

Browse Source 
PiperOrigin-RevId: 167995887
This commit is contained in:
A. Unique TensorFlower
2017-09-08 07:45:44 -07:00
committed by TensorFlower Gardener
parent 10b966b95d
commit 24cbb2a8ce
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
configure.py
View File

@@ -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)