[FIX] Fix docker build error (vllm-project#1831) (vllm-project#1832)

Co-authored-by: Antoni Baum <antoni.baum@protonmail.com>
liu-cong · Nov 30, 2023 · f07c1ce · f07c1ce
1 parent 63b2206
commit f07c1ce
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -18,6 +18,11 @@ RUN --mount=type=cache,target=/root/.cache/pip \
 # image to build pytorch extensions
 FROM dev AS build
 
+# install build dependencies
+COPY requirements-build.txt requirements-build.txt
+RUN --mount=type=cache,target=/root/.cache/pip \
+    pip install -r requirements-build.txt
+
 # copy input files
 COPY csrc csrc
 COPY setup.py setup.py

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,8 +1,9 @@
 [build-system]
+# Should be mirrored in requirements-build.txt
 requires = [
     "ninja",
     "packaging",
-    "setuptools",
+    "setuptools >= 49.4.0",
     "torch >= 2.1.0",
     "wheel",
 ]

diff --git a/requirements-build.txt b/requirements-build.txt
@@ -0,0 +1,6 @@
+# Should be mirrored in pyproject.toml
+ninja
+packaging
+setuptools>=49.4.0
+torch>=2.1.0
+wheel