Install torchvision in release docker (#6795)

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