From 4e7c93a1764ff11cef18f539c8463af7fd3cd825 Mon Sep 17 00:00:00 2001 From: Jinzhe Zeng Date: Tue, 26 Mar 2024 23:57:26 -0400 Subject: [PATCH] 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 (cherry picked from commit 197612a001a8815411a1942d87e571cc05580498) --- backend/dynamic_metadata.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/backend/dynamic_metadata.py b/backend/dynamic_metadata.py index c0b91d9372..717fd65931 100644 --- 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", ], }