From 7b9177dea5ec4cfd062fa33720ff2d7071c1a283 Mon Sep 17 00:00:00 2001 From: Siyuan Liu Date: Mon, 25 Mar 2024 10:26:28 -0700 Subject: [PATCH] Install torchvision in release docker (#6795) --- infra/ansible/Dockerfile | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/infra/ansible/Dockerfile b/infra/ansible/Dockerfile index 53ef91f9432..f2f327044ca 100644 --- a/infra/ansible/Dockerfile +++ b/infra/ansible/Dockerfile @@ -20,8 +20,7 @@ ARG ansible_vars RUN ansible-playbook -vvv playbook.yaml -e "stage=release" -e "${ansible_vars}" --tags "install_deps" WORKDIR /tmp/wheels -COPY --from=build /src/pytorch/dist/*.whl ./ -COPY --from=build /src/pytorch/xla/dist/*.whl ./ +COPY --from=build /dist/*.whl ./ RUN echo "Installing the following wheels" && ls *.whl RUN pip install *.whl