Update setup.py: replace libcudart.so.12.0 with libcudart.so.12

microsoft · Nov 18, 2023 · 7d74d6b · 7d74d6b
1 parent 41f9379
commit 7d74d6b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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",