switch setup-7

Dockerfile

@@ -6,7 +6,7 @@ FROM ghcr.io/huggingface/text-generation-inference:3.3.4@sha256:91822fefc613d742
 RUN apt-get update && \
     DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
         build-essential git cmake ninja-build pkg-config curl ca-certificates gnupg software-properties-common \
-        python3-pip python3.11 python3.11-dev python3.11-venv python3.11-distutils tzdata && \
+        python3-pip python3.11 python3.11-dev python3.11-venv python3.11-distutils python3-dev tzdata && \
     # Preconfigure timezone to UTC
     ln -fs /usr/share/zoneinfo/Etc/UTC /etc/localtime && dpkg-reconfigure --frontend noninteractive tzdata && \
     # Add CUDA keyring and repository
@@ -34,10 +34,8 @@ RUN git clone --depth 1 --recursive https://github.com/BachiLi/diffvg.git /tmp/d
     sed -i 's/^\s*cmake_minimum_required(VERSION .*$/cmake_minimum_required(VERSION 3.5)/' /tmp/diffvg/pybind11/CMakeLists.txt && \
     mkdir /tmp/diffvg/build && cd /tmp/diffvg/build && \
     cmake .. \
-        -DPYTHON_EXECUTABLE=$(which python3) \
-        -DPython3_ROOT_DIR=$(python3 -c "import sys; print(sys.prefix)") \
-        -DPYTHON_LIBRARY=/usr/lib/x86_64-linux-gnu/libpython3.11.so \
-        -DPYTHON_INCLUDE_DIR=/usr/include/python3.11 \
+        -DPython3_EXECUTABLE=$(which python3) \
+        -DPython3_FIND_STRATEGY=LOCATION \
         -DCMAKE_BUILD_TYPE=Release \
         -DDIFFVG_CUDA=0 && \
     make -j$(nproc) && \