diff --git a/Containerfile b/Containerfile index 149ee48..4ed6b36 100644 --- a/Containerfile +++ b/Containerfile @@ -86,6 +86,7 @@ ENV LD_LIBRARY_PATH="" ENV OKP_CONTENT=$OKP_CONTENT ENV BUILD_OCP_DOCS=$BUILD_OCP_DOCS +USER 0 WORKDIR /rag-content RUN if [ "$FLAVOR" == "gpu" ]; then \ diff --git a/temporary_cpu.containerfile b/temporary_cpu.containerfile index a5051ca..9126204 100644 --- a/temporary_cpu.containerfile +++ b/temporary_cpu.containerfile @@ -109,6 +109,7 @@ ENV LD_LIBRARY_PATH="" ENV OKP_CONTENT=$OKP_CONTENT ENV BUILD_OCP_DOCS=$BUILD_OCP_DOCS +USER 0 WORKDIR /rag-content RUN if [ "$BUILD_UPSTREAM_DOCS" = "true" ]; then \ diff --git a/temporary_gpu.containerfile b/temporary_gpu.containerfile index 85ff141..cda46c8 100644 --- a/temporary_gpu.containerfile +++ b/temporary_gpu.containerfile @@ -137,6 +137,7 @@ ENV LD_LIBRARY_PATH="" ENV OKP_CONTENT=$OKP_CONTENT ENV BUILD_OCP_DOCS=$BUILD_OCP_DOCS +USER 0 WORKDIR /rag-content RUN python -c "import torch, sys; available=torch.cuda.is_available(); print(f'CUDA is available: {available}'); sys.exit(0 if available else 1)"; \