experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Shorten find_cuda_config finder

Browse Source 
PiperOrigin-RevId: 353905388
Change-Id: Iaeed674c38bc9346d26e4701f413163212217c37
This commit is contained in:
Austin Anderson 2021-01-26 11:07:14 -08:00 committed by TensorFlower Gardener
parent f5d89fe581
commit b1f946aed5
1 changed files with 6 additions and 12 deletions

14
configure.py
View File

@ -20,8 +20,8 @@ from __future__ import print_function
import argparse import argparse
import errno import errno
import glob
import os import os
import pathlib
import platform import platform
import re import re
import subprocess import subprocess
@ -1239,18 +1239,12 @@ def validate_cuda_config(environ_cp):
if environ_cp.get('TF_NCCL_VERSION', None): if environ_cp.get('TF_NCCL_VERSION', None):
cuda_libraries.append('nccl') cuda_libraries.append('nccl')
find_cuda_path = pathlib.Path('third_party/gpus/find_cuda_config.py') paths = glob.glob('**/third_party/gpus/find_cuda_config.py', recursive=True)
if not pathlib.Path(find_cuda_path).is_file(): if not paths:
find_cuda_path = pathlib.Path('.').glob('**/' + str(find_cuda_path))
try:
find_cuda_path = find_cuda_path.__next__()
except StopIteration:
raise FileNotFoundError( raise FileNotFoundError(
"Can't find 'find_cuda_config.py' script inside working directory") "Can't find 'find_cuda_config.py' script inside working directory")
proc = subprocess.Popen( proc = subprocess.Popen(
[environ_cp['PYTHON_BIN_PATH'], [environ_cp['PYTHON_BIN_PATH'], paths[0]] + cuda_libraries,
str(find_cuda_path)] + cuda_libraries,
stdout=subprocess.PIPE, stdout=subprocess.PIPE,
env=maybe_encode_env(environ_cp)) env=maybe_encode_env(environ_cp))