Skip to content

Commit

Permalink
wip
Browse files Browse the repository at this point in the history
  • Loading branch information
malcolmgreaves committed Nov 27, 2024
1 parent a65294e commit 2a36612
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion ci/docker/Dockerfile_cuda_12.3
Original file line number Diff line number Diff line change
Expand Up @@ -116,7 +116,7 @@ EOF
WORKDIR /build/
# the --index-url **MUST MATCH** the base image CUDA version !!!
# maybe --no-build-isolation ??
RUN pip install --index-url https://download.pytorch.org/whl/cu123 torch==${PYTORCH_VERSION}
RUN pip install --index-url https://download.pytorch.org/whl/cu121 torch==${PYTORCH_VERSION}


FROM pytorch_base AS dl_deps_base
Expand Down

0 comments on commit 2a36612

Please sign in to comment.