diff --git a/setup.py b/setup.py
index b26a34f371..3ba8352f0b 100644
--- a/setup.py
+++ b/setup.py
@@ -631,7 +631,7 @@ def run(self):
             ),
             extra_compile_args=(
                 [
-                    f'/DPYBIND11_BUILD_ABI=\\"{torch._C._PYBIND11_BUILD_ABI}\\"',
+                    f'/DPYBIND11_BUILD_ABI=\\"{getattr(torch._C, "_PYBIND11_BUILD_ABI", "_cxxabi1018")}\\"',
                     "/GS-",
                     "/permissive-",
                     "/utf-8",