From 76740307173804204937271a896a8844f622b0b9 Mon Sep 17 00:00:00 2001 From: Siyuan Liu Date: Tue, 23 Apr 2024 22:01:06 -0700 Subject: [PATCH] Reland Install torchvision in release docker #6795 (#6860) (#6955) 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