@@ -138,7 +138,7 @@ WORKDIR /workspace
138138COPY requirements/common.txt requirements/common.txt
139139COPY requirements/cuda.txt requirements/cuda.txt
140140RUN --mount=type=cache,target=/root/.cache/uv \
141- uv pip install --system -r requirements/cuda.txt \
141+ uv pip install --python python3 -r requirements/cuda.txt \
142142 --extra-index-url ${PYTORCH_CUDA_INDEX_BASE_URL}/cu$(echo $CUDA_VERSION | cut -d. -f1,2 | tr -d '.' )
143143
144144# cuda arch list used by torch
@@ -168,7 +168,7 @@ ENV UV_INDEX_STRATEGY="unsafe-best-match"
168168ENV UV_LINK_MODE=copy
169169
170170RUN --mount=type=cache,target=/root/.cache/uv \
171- uv pip install --system -r requirements/build.txt \
171+ uv pip install --python python3 -r requirements/build.txt \
172172 --extra-index-url ${PYTORCH_CUDA_INDEX_BASE_URL}/cu$(echo $CUDA_VERSION | cut -d. -f1,2 | tr -d '.' )
173173
174174COPY . .
@@ -265,7 +265,7 @@ COPY requirements/lint.txt requirements/lint.txt
265265COPY requirements/test.txt requirements/test.txt
266266COPY requirements/dev.txt requirements/dev.txt
267267RUN --mount=type=cache,target=/root/.cache/uv \
268- uv pip install --system -r requirements/dev.txt \
268+ uv pip install --python python3 -r requirements/dev.txt \
269269 --extra-index-url ${PYTORCH_CUDA_INDEX_BASE_URL}/cu$(echo $CUDA_VERSION | cut -d. -f1,2 | tr -d '.' )
270270# ################### DEV IMAGE ####################
271271
0 commit comments