[chore] update torch version to 2.9 (#12969)

Co-authored-by: Baizhou Zhang <sobereddiezhang@gmail.com>
2025-11-25 14:47:34 -08:00
parent 64a11303ce
commit 36b1bcd242
13 changed files with 44 additions and 87 deletions
@@ -134,6 +134,8 @@ if [ "$IS_BLACKWELL" != "1" ]; then
    git clone --branch v0.5 --depth 1 https://github.com/EvolvingLMMs-Lab/lmms-eval.git
    $PIP_CMD install -e lmms-eval/ $PIP_INSTALL_SUFFIX
 fi
+# Cudnn with version less than 9.16.0.29 will cause performance regression on Conv3D kernel
+$PIP_CMD install nvidia-cudnn-cu12==9.16.0.29 --force-reinstall $PIP_INSTALL_SUFFIX
 $PIP_CMD uninstall xformers || true
 # Show current packages
 $PIP_CMD list