From b0ba29f98a695671972d4a4cc07441014dba2892 Mon Sep 17 00:00:00 2001 From: Siyuan Liu Date: Fri, 29 Mar 2024 16:20:36 -0700 Subject: [PATCH] Reland Install torchvision in release docker #6795 (#6860) Co-authored-by: Siyuan Liu --- infra/ansible/Dockerfile | 1 + 1 file changed, 1 insertion(+) diff --git a/infra/ansible/Dockerfile b/infra/ansible/Dockerfile index 53ef91f9432..acd726061ad 100644 --- a/infra/ansible/Dockerfile +++ b/infra/ansible/Dockerfile @@ -22,6 +22,7 @@ RUN ansible-playbook -vvv playbook.yaml -e "stage=release" -e "${ansible_vars}" WORKDIR /tmp/wheels COPY --from=build /src/pytorch/dist/*.whl ./ COPY --from=build /src/pytorch/xla/dist/*.whl ./ +COPY --from=build /dist/torchvision*.whl ./ RUN echo "Installing the following wheels" && ls *.whl RUN pip install *.whl