Skip to content

Commit

Permalink
revert
Browse files Browse the repository at this point in the history
  • Loading branch information
malcolmgreaves committed Dec 6, 2024
1 parent 3274183 commit 5e1f681
Showing 1 changed file with 1 addition and 11 deletions.
12 changes: 1 addition & 11 deletions ci/docker/Dockerfile_cuda_12.3
Original file line number Diff line number Diff line change
Expand Up @@ -120,17 +120,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/cu121 torch==${PYTORCH_VERSION}
RUN <<EOF
set -e
# get version
git clone --recursive https://github.com/pytorch/pytorch
cd pytorch
git checkout v${PYTORCH_VERSION}
git submodule sync
git submodule update --init --recursive
# build
EOF
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 5e1f681

Please sign in to comment.