Add cudnn 9 dlls

microsoft · Aug 14, 2024 · 9416e17 · 9416e17
1 parent a0708a0
commit 9416e17
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -208,6 +208,16 @@ def run(self):
                     "libcufft.so.10",
                     "libcufft.so.11",
                     "libcurand.so.10",
+                    "libcudnn_adv.so.9",
+                    "libcudnn_cnn.so.9",
+                    "libcudnn_engines_precompiled.so.9",
+                    "libcudnn_engines_runtime_compiled.so.9",
+                    "libcudnn_graph.so.9",
+                    "libcudnn_heuristic.so.9",
+                    "libcudnn_ops.so.9",
+                    "libnvJitLink.so.12",
+                    "libnvrtc.so.12",
+                    "libnvrtc-builtins.so.12",
                 ]
                 rocm_dependencies = [
                     "libamd_comgr.so.2",