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

Co-authored-by: Antoni Baum <antoni.baum@protonmail.com>
This commit is contained in:
Allen
2023-11-30 15:06:50 +08:00
committed by GitHub
parent 63b2206ad0
commit f07c1ceaa5
3 changed files with 13 additions and 1 deletions

View File

@ -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

View File

@ -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",
]

6
requirements-build.txt Normal file
View File

@ -0,0 +1,6 @@
# Should be mirrored in pyproject.toml
ninja
packaging
setuptools>=49.4.0
torch>=2.1.0
wheel