From defd20a5f4714337bc2cc232eba3049db2b8c367 Mon Sep 17 00:00:00 2001 From: Siyuan Liu Date: Thu, 21 Mar 2024 04:49:46 +0000 Subject: [PATCH] copy torchvision whl --- infra/ansible/Dockerfile | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/infra/ansible/Dockerfile b/infra/ansible/Dockerfile index 53ef91f9432..f2f327044ca 100644 --- 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