Skip to content

Commit fcf4117

Browse files
committed
build: não força cpu em torch
1 parent e8293a6 commit fcf4117

File tree

3 files changed

+291
-27
lines changed

3 files changed

+291
-27
lines changed

Dockerfile

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,8 @@ RUN pip install poetry \
77
&& poetry config virtualenvs.create false \
88
&& poetry install --only main --no-root --no-directory
99
COPY . /dptoie_python
10+
11+
RUN pip install torch==2.7.1+cpu -f https://download.pytorch.org/whl/torch_stable.html
1012
RUN poetry install --only main --no-root
1113

1214
ENV PYTHONPATH="$PYTHONPATH:/dptoie_python"

0 commit comments

Comments
 (0)