pin nvidia-cudnn-cu{11,12} to <9 (#3610)

`nvidia-cudnn-cu12` 9.0.0.312 seems broken (and might not be compatible) Signed-off-by: Jinzhe Zeng <jinzhe.zeng@rutgers.edu> (cherry picked from commit 197612a)
deepmodeling · Apr 6, 2024 · 8c0f298 · 8c0f298
1 parent 62c1315
commit 8c0f298
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/backend/dynamic_metadata.py b/backend/dynamic_metadata.py
@@ -75,7 +75,7 @@ def dynamic_metadata(
                 "nvidia-curand-cu11",
                 "nvidia-cusolver-cu11",
                 "nvidia-cusparse-cu11",
-                "nvidia-cudnn-cu11",
+                "nvidia-cudnn-cu11<9",
                 "nvidia-cuda-nvcc-cu11",
             ],
             "cu12": [
@@ -85,7 +85,7 @@ def dynamic_metadata(
                 "nvidia-curand-cu12",
                 "nvidia-cusolver-cu12",
                 "nvidia-cusparse-cu12",
-                "nvidia-cudnn-cu12",
+                "nvidia-cudnn-cu12<9",
                 "nvidia-cuda-nvcc-cu12",
             ],
         }