switch setup-6

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 tzdata && \
+        python3-pip python3.11 python3.11-dev python3.11-venv python3.11-distutils tzdata && \
     # Preconfigure timezone to UTC
     ln -fs /usr/share/zoneinfo/Etc/UTC /etc/localtime && dpkg-reconfigure --frontend noninteractive tzdata && \
     # Add CUDA keyring and repository
@@ -32,7 +32,15 @@ RUN python3 -m pip install --upgrade pip && \
 RUN git clone --depth 1 --recursive https://github.com/BachiLi/diffvg.git /tmp/diffvg && \
     # Lower cmake requirement for pybind11
     sed -i 's/^\s*cmake_minimum_required(VERSION .*$/cmake_minimum_required(VERSION 3.5)/' /tmp/diffvg/pybind11/CMakeLists.txt && \
-    cd /tmp/diffvg && \
-    python3 setup.py install && \
+    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 \
+        -DCMAKE_BUILD_TYPE=Release \
+        -DDIFFVG_CUDA=0 && \
+    make -j$(nproc) && \
+    make install && \
     cd / && rm -rf /tmp/diffvg