Reland Install torchvision in release docker #6795 (#6860) (#6955)

Co-authored-by: Siyuan Liu <lsiyuan@google.coim>
pytorch · Apr 24, 2024 · 7674030 · 7674030
1 parent 2ebd636
commit 7674030
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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