Skip to content

Commit c6b53a6

Browse files
committed
Add CUDA 12.x support for torch
1 parent a4d4e2e commit c6b53a6

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
@@ -36,7 +36,7 @@ ENV PATH="$VIRTUAL_ENV/bin:$PATH"
3636

3737
COPY requirements.txt requirements.txt
3838
RUN uv pip install --upgrade pip
39-
RUN uv pip install -r requirements.txt
39+
RUN uv pip install -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cu128 --index-strategy unsafe-best-match
4040

4141
WORKDIR /app
4242

0 commit comments

Comments
 (0)