From 0bc5a76559cd042f3b09baffd86b228f5561549a Mon Sep 17 00:00:00 2001 From: Rostan Tabet Date: Tue, 1 Apr 2025 18:09:46 +0200 Subject: [PATCH] fix: use CMake>=3.5 when building cuDNN FE cuDNN FE depends on dlpack 0.8, which uses CMake 3.2. CMake 4.0 dropped support for CMake<3.5. This shouldn't break anything in DLPack as a similar fix was done recently upstream (https://github.com/dmlc/dlpack/pull/163). Signed-off-by: Rostan Tabet --- docker/Dockerfile | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/docker/Dockerfile b/docker/Dockerfile index ad53083..9335bc2 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -110,7 +110,8 @@ RUN git clone -b release-v3 https://github.com/NVIDIA/NVTX.git && cd NVTX/python python3 -m pip install --no-build-isolation . # Install cuDNN FE -RUN apt install -y cudnn && export CUDNN_INCLUDE_DIR=/usr/include && \ +RUN apt install -y cudnn && \ + CUDNN_INCLUDE_DIR=/usr/include CMAKE_POLICY_VERSION_MINIMUM=3.5 \ python3 -m pip install git+https://github.com/NVIDIA/cudnn-frontend.git # Install CUDA-Python