diff --git a/setup.py b/setup.py index 5e69fa50f52..829b9c2f910 100644 --- a/setup.py +++ b/setup.py @@ -6,6 +6,7 @@ import shutil import subprocess import sys +import sysconfig import warnings from pathlib import Path @@ -136,6 +137,8 @@ def get_macros_and_flags(): if sys.platform == "win32": define_macros += [("torchvision_EXPORTS", None)] extra_compile_args["cxx"].append("/MP") + if sysconfig.get_config_var("Py_GIL_DISABLED"): + extra_compile_args["cxx"].append("-DPy_GIL_DISABLED") if DEBUG: extra_compile_args["cxx"].append("-g")