From 6ca964b41090935e7aa74cdbcfe09a891130167f Mon Sep 17 00:00:00 2001 From: atalman Date: Thu, 21 Sep 2023 17:07:45 +0000 Subject: [PATCH] Remove torchtext from Build Official Docker images (#109799) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Fixes nightly official Docker image build. Failures: https://hud.pytorch.org/hud/pytorch/pytorch/nightly/1?per_page=50&name_filter=Build%20Official ### 🤖 Generated by Copilot at 8671bfc Remove `torchtext` installation from `Dockerfile` for arm64. This fixes the arm64 build of the PyTorch Docker image. Pull Request resolved: https://github.com/pytorch/pytorch/pull/109799 Approved by: https://github.com/seemethere --- Dockerfile | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/Dockerfile b/Dockerfile index 7bc6b200e87f9..62a5b839a1b56 100644 --- a/Dockerfile +++ b/Dockerfile @@ -73,8 +73,8 @@ ARG TARGETPLATFORM # On arm64 we can only install wheel packages. RUN case ${TARGETPLATFORM} in \ - "linux/arm64") pip install --extra-index-url https://download.pytorch.org/whl/cpu/ torch torchvision torchaudio torchtext ;; \ - *) /opt/conda/bin/conda install -c "${INSTALL_CHANNEL}" -c "${CUDA_CHANNEL}" -y "python=${PYTHON_VERSION}" pytorch torchvision torchaudio torchtext "pytorch-cuda=$(echo $CUDA_VERSION | cut -d'.' -f 1-2)" ;; \ + "linux/arm64") pip install --extra-index-url https://download.pytorch.org/whl/cpu/ torch torchvision torchaudio ;; \ + *) /opt/conda/bin/conda install -c "${INSTALL_CHANNEL}" -c "${CUDA_CHANNEL}" -y "python=${PYTHON_VERSION}" pytorch torchvision torchaudio "pytorch-cuda=$(echo $CUDA_VERSION | cut -d'.' -f 1-2)" ;; \ esac && \ /opt/conda/bin/conda clean -ya RUN /opt/conda/bin/pip install torchelastic