From 6994120454b5683cba0dc23625fb31509e144eac Mon Sep 17 00:00:00 2001 From: Kaniska Date: Fri, 14 Feb 2025 09:14:54 +0000 Subject: [PATCH] nvidia-cuda fix for Issue #1270 --- src/nvidia-cuda/install.sh | 2 +- test/nvidia-cuda/install_cuda_12_3_version.sh | 4 ++-- test/nvidia-cuda/install_cuda_12_4_version.sh | 4 ++-- 3 files changed, 5 insertions(+), 5 deletions(-) diff --git a/src/nvidia-cuda/install.sh b/src/nvidia-cuda/install.sh index 1cfc7668e..a1055ac45 100644 --- a/src/nvidia-cuda/install.sh +++ b/src/nvidia-cuda/install.sh @@ -74,7 +74,7 @@ if [ "$CUDNN_VERSION" = "automatic" ]; then if [[ "$CUDA_VERSION" < "12.3" ]]; then CUDNN_VERSION=$(apt-cache policy libcudnn8 | grep "$CUDA_VERSION" | grep -Eo '^[^-1+]*' | sort -V | tail -n1 | xargs) else - CUDNN_VERSION=$(apt-cache policy libcudnn9-cuda-$major_cuda_version | grep "Candidate" | awk '{print $2}' | grep -Eo '^[^-1+]*') + CUDNN_VERSION=$(apt-cache policy libcudnn9-cuda-$major_cuda_version | grep "Candidate" | awk '{print $2}' | grep -Eo '^[^-+]*') fi fi major_cudnn_version=$(echo "${CUDNN_VERSION}" | cut -d '.' -f 1) diff --git a/test/nvidia-cuda/install_cuda_12_3_version.sh b/test/nvidia-cuda/install_cuda_12_3_version.sh index 95d4e8bdc..15c2abd7a 100644 --- a/test/nvidia-cuda/install_cuda_12_3_version.sh +++ b/test/nvidia-cuda/install_cuda_12_3_version.sh @@ -5,8 +5,8 @@ set -e # Optional: Import test library source dev-container-features-test-lib -# # Check installation of libcudnn9-cuda-12 (9.4.0) -check "libcudnn.so.9.5.0" test 1 -eq "$(find /usr -name 'libcudnn.so.9.5.0' | wc -l)" +# Check installation of libcudnn9 +check "libcudnn.so.9" test 1 -eq "$(find /usr -name 'libcudnn.so.9' | wc -l)" # Check installation of cuda-nvtx-12-3 (12.3) check "cuda-12-3+nvtx" test -e '/usr/local/cuda-12.3/targets/x86_64-linux/include/nvtx3/' diff --git a/test/nvidia-cuda/install_cuda_12_4_version.sh b/test/nvidia-cuda/install_cuda_12_4_version.sh index 77ecbf190..284920a33 100644 --- a/test/nvidia-cuda/install_cuda_12_4_version.sh +++ b/test/nvidia-cuda/install_cuda_12_4_version.sh @@ -5,8 +5,8 @@ set -e # Optional: Import test library source dev-container-features-test-lib -# # Check installation of libcudnn9-cuda-12 (9.4.0) -check "libcudnn.so.9.5.0" test 1 -eq "$(find /usr -name 'libcudnn.so.9.5.0' | wc -l)" +# Check installation of libcudnn9 +check "libcudnn.so.9" test 1 -eq "$(find /usr -name 'libcudnn.so.9' | wc -l)" # Check installation of cuda-nvtx-12-4 (12.4) check "cuda-12-4+nvtx" test -e '/usr/local/cuda-12.4/targets/x86_64-linux/include/nvtx3/'