Skip to content

Commit 4bf853f

Browse files
committed
fix: run torch sanity check in venv
1 parent 07fee49 commit 4bf853f

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ RUN --mount=type=cache,target=$POETRY_CACHE_DIR \
1414
poetry install --without dev --no-root
1515

1616
# Sanity check: ensure torch is CPU-only (cuda should be None)
17-
RUN python -c "import torch; print('torch', torch.__version__); print('torch.cuda', torch.version.cuda); print('cuda available', torch.cuda.is_available())"
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())"
1818

1919

2020
FROM ghcr.io/lambda-feedback/evaluation-function-base/python:3.12

0 commit comments

Comments
 (0)