Merge branch 'freeze-version'

benz0li · benz0li · commit 52958597e14b · 2025-06-26T15:45:34.000+02:00
diff --git a/cuda/3.13.4.Dockerfile b/cuda/3.13.4.Dockerfile
@@ -102,10 +102,13 @@ LABEL com.nvidia.cudnn.version="${CUDNN_VERSION_AMD64}"
 ENV CUDNN_VERSION=${CUDNN_VERSION_AMD64}
 ENV NV_CUDNN_PACKAGE_VERSION=${CUDNN_VERSION_AMD64}
 ENV NV_CUDNN_DEV_PACKAGE_VERSION=${CUDNN_VERSION_AMD64}
+ENV NV_CUDNN_HEADERS_PACKAGE_VERSION=${CUDNN_VERSION_AMD64}
 ENV NV_CUDNN_PACKAGE_NAME=libcudnn${NV_CUDNN_PACKAGE_VERSION%%.*}-cuda-${CUDNN_CUDA_VERSION_MAJ_MIN_AMD64%%.*}
 ENV NV_CUDNN_DEV_PACKAGE_NAME=libcudnn${NV_CUDNN_DEV_PACKAGE_VERSION%%.*}-dev-cuda-${CUDNN_CUDA_VERSION_MAJ_MIN_AMD64%%.*}
+ENV NV_CUDNN_HEADERS_PACKAGE_NAME=libcudnn${NV_CUDNN_HEADERS_PACKAGE_VERSION%%.*}-headers-cuda-${CUDNN_CUDA_VERSION_MAJ_MIN_AMD64%%.*}
 ENV NV_CUDNN_PACKAGE="${NV_CUDNN_PACKAGE_NAME}=${NV_CUDNN_PACKAGE_VERSION}-1"
 ENV NV_CUDNN_DEV_PACKAGE="${NV_CUDNN_DEV_PACKAGE_NAME}=${NV_CUDNN_DEV_PACKAGE_VERSION}-1"
+ENV NV_CUDNN_HEADERS_PACKAGE="${NV_CUDNN_HEADERS_PACKAGE_NAME}=${NV_CUDNN_HEADERS_PACKAGE_VERSION}-1"
 
 FROM ${BUILD_ON_IMAGE} AS cudnn9-devel-arm64
 
@@ -117,10 +120,13 @@ LABEL com.nvidia.cudnn.version="${CUDNN_VERSION_ARM64}"
 ENV CUDNN_VERSION=${CUDNN_VERSION_ARM64}
 ENV NV_CUDNN_PACKAGE_VERSION=${CUDNN_VERSION_ARM64}
 ENV NV_CUDNN_DEV_PACKAGE_VERSION=${CUDNN_VERSION_ARM64}
+ENV NV_CUDNN_HEADERS_PACKAGE_VERSION=${CUDNN_VERSION_ARM64}
 ENV NV_CUDNN_PACKAGE_NAME=libcudnn${NV_CUDNN_PACKAGE_VERSION%%.*}-cuda-${CUDNN_CUDA_VERSION_MAJ_MIN_ARM64%%.*}
 ENV NV_CUDNN_DEV_PACKAGE_NAME=libcudnn${NV_CUDNN_DEV_PACKAGE_VERSION%%.*}-dev-cuda-${CUDNN_CUDA_VERSION_MAJ_MIN_ARM64%%.*}
+ENV NV_CUDNN_HEADERS_PACKAGE_NAME=libcudnn${NV_CUDNN_HEADERS_PACKAGE_VERSION%%.*}-headers-cuda-${CUDNN_CUDA_VERSION_MAJ_MIN_ARM64%%.*}
 ENV NV_CUDNN_PACKAGE="${NV_CUDNN_PACKAGE_NAME}=${NV_CUDNN_PACKAGE_VERSION}-1"
 ENV NV_CUDNN_DEV_PACKAGE="${NV_CUDNN_DEV_PACKAGE_NAME}=${NV_CUDNN_DEV_PACKAGE_VERSION}-1"
+ENV NV_CUDNN_HEADERS_PACKAGE="${NV_CUDNN_HEADERS_PACKAGE_NAME}=${NV_CUDNN_HEADERS_PACKAGE_VERSION}-1"
 
 FROM cudnn${CUDNN_VERSION_MAJ}-${CUDA_IMAGE_FLAVOR}-${TARGETARCH}
 
@@ -168,9 +174,11 @@ RUN cpuBlasLib="$(update-alternatives --query \
   && apt-get -y install --no-install-recommends \
     ${NV_CUDNN_PACKAGE} \
     ${NV_CUDNN_DEV_PACKAGE} \
+    ${NV_CUDNN_HEADERS_PACKAGE} \
   ## Keep apt from auto upgrading the cuDNN packages
   && apt-mark hold \
     ${NV_CUDNN_PACKAGE_NAME} \
     ${NV_CUDNN_DEV_PACKAGE_NAME} \
+    ${NV_CUDNN_HEADERS_PACKAGE_NAME} \
   ## Clean up
   && rm -rf /var/lib/apt/lists/*
