Skip to content

Commit c8f22be

Browse files
committed
LCORE-438: Add detection of GPU Python libs in image building
1 parent 8d7a744 commit c8f22be

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

Containerfile-gpu

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -38,8 +38,8 @@ RUN uv lock && uv sync --locked --no-install-project
3838
# Installing separately from its dependencies allows optimal layer caching
3939
RUN uv sync --locked
4040

41-
# Test PyTorch was installed with default GPU linux backend (CUDA)
42-
#RUN uv run python -c "import torch; print(torch.__version__); assert 'cpu' not in torch.__version__;"
41+
# Test PyTorch was installed with GPU libraries (CUDA)
42+
RUN uv run python -c "from nvidia import cuda_runtime"
4343

4444
# Add executables from .venv to system PATH
4545
ENV PATH="/rag-content/.venv/bin:$PATH"

0 commit comments

Comments
 (0)