diff --git a/tensorflow/python/platform/sysconfig.py b/tensorflow/python/platform/sysconfig.py index 13eda341896568..d12223e00cd18c 100644 --- a/tensorflow/python/platform/sysconfig.py +++ b/tensorflow/python/platform/sysconfig.py @@ -19,6 +19,7 @@ from __future__ import print_function import os.path as _os_path +import platform as _platform from tensorflow.python.framework.versions import CXX11_ABI_FLAG as _CXX11_ABI_FLAG from tensorflow.python.framework.versions import MONOLITHIC_BUILD as _MONOLITHIC_BUILD @@ -73,8 +74,13 @@ def get_link_flags(): Returns: The link flags. """ + is_mac = _platform.system() == 'Darwin' + ver = _VERSION.split('.')[0] flags = [] if not _MONOLITHIC_BUILD: flags.append('-L%s' % get_lib()) - flags.append('-l:libtensorflow_framework.so.%s' % _VERSION.split('.')[0]) + if is_mac: + flags.append('-l:libtensorflow_framework.%s.dylib' % ver) + else: + flags.append('-l:libtensorflow_framework.so.%s' % ver) return flags