Skip to content

Commit 12f2d9b

Browse files
committed
fix
1 parent 5ed03e3 commit 12f2d9b

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

Dockerfile

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,10 @@ RUN chmod 777 -R /tmp && apt-get update && DEBIAN_FRONTEND=noninteractive apt-ge
1313
curl \
1414
g++ \
1515
make \
16-
git && \
16+
git \
17+
cmake \
18+
ninja-build \
19+
build-essential && \
1720
rm -rf /var/lib/apt/lists/*
1821

1922
RUN case ${TARGETPLATFORM} in \
@@ -34,7 +37,7 @@ RUN case ${TARGETPLATFORM} in \
3437

3538
WORKDIR /root
3639

37-
RUN pip install torch==2.7.1
40+
RUN pip install torch==2.7.1 numpy
3841

3942
COPY . .
4043

0 commit comments

Comments
 (0)