diff --git a/configure.py b/configure.py index fea37d30a9..1d173ef45b 100644 --- a/configure.py +++ b/configure.py @@ -44,6 +44,10 @@ def is_windows(): return platform.system() == "Windows" +def is_raspi_arm(): + return os.uname()[4] == "armv7l" + + def get_tf_header_dir(): import tensorflow as tf @@ -60,6 +64,8 @@ def get_tf_shared_lib_dir(): if is_windows(): tf_shared_lib_dir = tf.sysconfig.get_compile_flags()[0][2:-7] + "python" return tf_shared_lib_dir.replace("\\", "/") + elif is_raspi_arm(): + return tf.sysconfig.get_compile_flags()[0][2:-7] + "python" else: return tf.sysconfig.get_link_flags()[0][2:] @@ -75,6 +81,9 @@ def get_shared_lib_name(): elif is_windows(): # Windows return "_pywrap_tensorflow_internal.lib" + elif is_raspi_arm(): + # The below command for linux would return an empty list + return "_pywrap_tensorflow_internal.so" else: # Linux return namespec[1][3:]