Revert "Install torchvision in release docker (#6795)"

This reverts commit 7b9177d.
pytorch · Mar 26, 2024 · 46f5b86 · 46f5b86
1 parent 899a0fa
commit 46f5b86
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/infra/ansible/Dockerfile b/infra/ansible/Dockerfile
@@ -20,7 +20,8 @@ ARG ansible_vars
 RUN ansible-playbook -vvv playbook.yaml -e "stage=release" -e "${ansible_vars}" --tags "install_deps"
 
 WORKDIR /tmp/wheels
-COPY --from=build /dist/*.whl ./
+COPY --from=build /src/pytorch/dist/*.whl ./
+COPY --from=build /src/pytorch/xla/dist/*.whl ./
 
 RUN echo "Installing the following wheels" && ls *.whl
 RUN pip install *.whl