diff --git a/setup.py b/setup.py index 477d187a2..f2ff878b4 100644 --- a/setup.py +++ b/setup.py @@ -14,6 +14,7 @@ import shutil import struct import subprocess import sys +import sysconfig import warnings from collections.abc import Iterator @@ -565,8 +566,16 @@ class pil_build_ext(build_ext): 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