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

Co-authored-by: Siyuan Liu <[email protected]>
pytorch · Mar 29, 2024 · b0ba29f · b0ba29f
1 parent 23aa4ef
commit b0ba29f
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