From dcbabe96f03ef5fef9116704a283b586c82057c5 Mon Sep 17 00:00:00 2001
From: "Yu-Hang \"Maxin\" Tang" <Tang.Maxin@gmail.com>
Date: Mon, 24 Feb 2025 10:14:35 -0800
Subject: [PATCH] Update symlnk-cudnn.sh

libcudnn is renamed to nvidia-cudnn in CUDA 12.9
---
 .github/container/symlnk-cudnn.sh | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/.github/container/symlnk-cudnn.sh b/.github/container/symlnk-cudnn.sh
index 5db2c411f..e961ea320 100755
--- a/.github/container/symlnk-cudnn.sh
+++ b/.github/container/symlnk-cudnn.sh
@@ -14,13 +14,13 @@ if [[ -d "${prefix}" ]]; then
 fi
 
 arch=$(uname -m)-linux-gnu
-libcudnn_pkgs=$(dpkg -l 'libcudnn*' | awk '/^ii/ {print $2}')
-if [[ -z "${libcudnn_pkgs}" ]]; then
-  echo "No libcudnn packages installed."
+cudnn_pkgs=$(dpkg -l '*cudnn*' | awk '/^ii/ {print $2}')
+if [[ -z "${cudnn_pkgs}" ]]; then
+  echo "No cuDNN packages installed."
   exit 1
 fi
 
-for cudnn_file in $(dpkg -L ${libcudnn_pkgs} | sort -u); do
+for cudnn_file in $(dpkg -L ${cudnn_pkgs} | sort -u); do
   # Real files and symlinks are linked into $prefix
   if [[ -f "${cudnn_file}" || -h "${cudnn_file}" ]]; then
     # Replace /usr with $prefix
@@ -38,4 +38,4 @@ for cudnn_file in $(dpkg -L ${libcudnn_pkgs} | sort -u); do
   else
     echo "Skipping ${cudnn_file}"
   fi
-done
\ No newline at end of file
+done