pin cudnn version to prevent perf regression

ardenma · ardenma · commit 8ab44672350c · 2026-01-17T16:27:54.000-08:00
diff --git a/pyproject.toml b/pyproject.toml
@@ -14,6 +14,7 @@ dependencies = [
     "torchvision==0.24.1",
     "transformers==4.57.1",
     "qwen-vl-utils==0.0.11",
+    "nvidia-cudnn-cu12==9.16.0.29", # https://github.com/pytorch/pytorch/issues/168167
     "datasets",
     "setuptools",
     "tqdm",
