diff --git a/setup.py b/setup.py index d26ca8bcd..6d24faac3 100644 --- a/setup.py +++ b/setup.py @@ -799,7 +799,10 @@ def pkgconfig_win(flag, cut): pass else: ext_args['extra_compile_args'].append("-std=c++11") - ext_args['library_dirs'].append("/usr/lib/x86_64-linux-gnu/") + if platform.machine() == 'x86_64': + ext_args['library_dirs'].append("/usr/lib/x86_64-linux-gnu/") + elif platform.machine() == 'aarch64': + ext_args['library_dirs'].append("/usr/lib/aarch64-linux-gnu/") # gcc? use standard library # ext_args['extra_compile_args'].append("-stdlib=libstdc++") # ext_args['extra_link_args'].append("-stdlib=libstdc++")