diff --git a/setup.py b/setup.py index 1bbd2c05cc9..0ad9204d0da 100644 --- a/setup.py +++ b/setup.py @@ -14,6 +14,7 @@ import struct import subprocess import sys +import sysconfig import warnings from setuptools import Extension, setup @@ -503,8 +504,16 @@ def build_extensions(self): for d in os.environ[k].split(os.path.pathsep): _add_directory(library_dirs, d) - _add_directory(library_dirs, os.path.join(sys.prefix, "lib")) - _add_directory(include_dirs, os.path.join(sys.prefix, "include")) + _add_directory( + library_dirs, + (sys.prefix == sys.base_prefix and sysconfig.get_config_var("LIBDIR")) + or os.path.join(sys.prefix, "lib"), + ) + _add_directory( + include_dirs, + (sys.prefix == sys.base_prefix and sysconfig.get_config_var("INCLUDEDIR")) + or os.path.join(sys.prefix, "include"), + ) # # add platform directories