diff --git a/setup.py b/setup.py
index 1c04433c9a7ca..a4b59a32f2481 100644
--- a/setup.py
+++ b/setup.py
@@ -196,7 +196,7 @@ def run(self):
                     "libcublasLt.so.11",
                     "libcublasLt.so.12",
                     "libcudart.so.11.0",
-                    "libcudart.so.12.0",
+                    "libcudart.so.12",
                     "libcudnn.so.8",
                     "libcufft.so.10",
                     "libcufft.so.11",