Skip to content

Commit 2473a31

Browse files
committed
fixup
Signed-off-by: Tyler Michael Smith <[email protected]>
1 parent a0fd1a5 commit 2473a31

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

docker/Dockerfile

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -138,7 +138,7 @@ WORKDIR /workspace
138138
COPY requirements/common.txt requirements/common.txt
139139
COPY requirements/cuda.txt requirements/cuda.txt
140140
RUN --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"
168168
ENV UV_LINK_MODE=copy
169169

170170
RUN --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

174174
COPY . .
@@ -265,7 +265,7 @@ COPY requirements/lint.txt requirements/lint.txt
265265
COPY requirements/test.txt requirements/test.txt
266266
COPY requirements/dev.txt requirements/dev.txt
267267
RUN --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

Comments
 (0)