diff --git a/mx_driving/csrc/CMakeLists.txt b/mx_driving/csrc/CMakeLists.txt
index 45259046ca109613f6fcb3ecff46cc496bdc2dce..44c0bd5f472c587d01c057de691c72d5095e39fb 100644
--- a/mx_driving/csrc/CMakeLists.txt
+++ b/mx_driving/csrc/CMakeLists.txt
@@ -49,7 +49,7 @@ if(BUILD_STAGE EQUAL 2)
             -fstack-protector-all
             -DTORCH_API_INCLUDE_EXTENSION_H
             -DTORCH_EXTENSION_NAME=_C
-            -D_GLIBCXX_USE_CXX11_ABI=0
+            -D_GLIBCXX_USE_CXX11_ABI=${ABI}
             -D__FILENAME__=__FILE__
             ${EXT_CXX_FLAGS})
 
diff --git a/setup.py b/setup.py
index 4356937e4491cdcd6e060b386c02494f89832df1..47b39b0455cc54261c4f13bbe7e4d551a43c6005 100644
--- a/setup.py
+++ b/setup.py
@@ -122,6 +122,11 @@ class ExtBuild(build_ext):
         ]
         if LooseVersion(torch.__version__) < LooseVersion("2.1.0"):
             cmake_args.append("-DCOMPILE_WITH_XLA:BOOL=ON")
+
+        if LooseVersion(torch.__version__) >= LooseVersion("2.6.0"):
+            cmake_args.append("-DABI=1")
+        else:
+            cmake_args.append("-DABI=0")
         build_args = ["--build", self.build_temp, f"-j{multiprocessing.cpu_count()}"]
 
         subprocess.check_call(