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

Checking that the NCCL 2 license file is found, see GitHub issue 19679.

Browse Source 
PiperOrigin-RevId: 202613754
This commit is contained in:
A. Unique TensorFlower 2018-06-29 01:44:59 -07:00 committed by TensorFlower Gardener
parent 6874e1ef40
commit f70dfea26f
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
configure.py
View File

@ -1134,7 +1134,9 @@ def set_tf_nccl_install_path(environ_cp):
nccl_lib_path = os.path.join(nccl_install_path, nccl_lib_path)
nccl_hdr_path = os.path.join(nccl_install_path, 'include/nccl.h')
if os.path.exists(nccl_lib_path) and os.path.exists(nccl_hdr_path):
nccl_license_path = os.path.join(nccl_install_path, 'NCCL-SLA.txt')
if os.path.exists(nccl_lib_path) and os.path.exists(
nccl_hdr_path) and os.path.exists(nccl_license_path):
# Set NCCL_INSTALL_PATH
environ_cp['NCCL_INSTALL_PATH'] = nccl_install_path
write_action_env_to_bazelrc('NCCL_INSTALL_PATH', nccl_install_path)