Skip to content

Commit 062567e

Browse files
committed
chore(docker): revert Dockerfile to base template + CPU-only torch
1 parent 4bf853f commit 062567e

1 file changed

Lines changed: 2 additions & 8 deletions

File tree

Dockerfile

Lines changed: 2 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
FROM ghcr.io/lambda-feedback/evaluation-function-base/python:3.12 AS builder
22

3-
RUN python -m pip install --no-cache-dir poetry==1.8.3
3+
RUN pip install poetry==1.8.3
44

55
ENV POETRY_NO_INTERACTION=1 \
66
POETRY_VIRTUALENVS_IN_PROJECT=1 \
@@ -9,14 +9,9 @@ ENV POETRY_NO_INTERACTION=1 \
99

1010
COPY pyproject.toml poetry.lock ./
1111

12-
# Install deps (torch/torchvision will be resolved from pytorch-cpu source via pyproject.toml)
1312
RUN --mount=type=cache,target=$POETRY_CACHE_DIR \
1413
poetry install --without dev --no-root
1514

16-
# Sanity check: ensure torch is CPU-only (cuda should be None)
17-
RUN /app/.venv/bin/python -c "import torch; print('torch', torch.__version__); print('torch.cuda', torch.version.cuda); print('cuda available', torch.cuda.is_available())"
18-
19-
2015
FROM ghcr.io/lambda-feedback/evaluation-function-base/python:3.12
2116

2217
ENV VIRTUAL_ENV=/app/.venv \
@@ -36,6 +31,5 @@ ENV FUNCTION_COMMAND="python"
3631
# Args to start the evaluation function with
3732
ENV FUNCTION_ARGS="-m,evaluation_function.main"
3833

39-
# Interface / logging
4034
ENV FUNCTION_INTERFACE="file"
41-
ENV LOG_LEVEL="debug"
35+
ENV LOG_LEVEL="debug"

0 commit comments

Comments
 (0)