From 2a46a9baff364b79fbbec4552ed98c31605d6475 Mon Sep 17 00:00:00 2001 From: Andrey Talman Date: Tue, 3 Dec 2024 12:02:17 -0500 Subject: [PATCH] Update unittest_ci.yml - Try using pypi --- .github/workflows/unittest_ci.yml | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/.github/workflows/unittest_ci.yml b/.github/workflows/unittest_ci.yml index 8865acee4..6fc71a014 100644 --- a/.github/workflows/unittest_ci.yml +++ b/.github/workflows/unittest_ci.yml @@ -73,9 +73,11 @@ jobs: conda info python --version conda run -n build_binary python --version - conda install -n build_binary \ - --yes \ - pytorch pytorch-cuda=11.8 -c pytorch-nightly -c nvidia + conda run -n build_binary \ + pip install torch --index-url https://download.pytorch.org/whl/nightly/cu118 + conda run -n build_binary \ + python -c "import torch" + echo "torch succeeded" conda run -n build_binary \ python -c "import torch.distributed" conda run -n build_binary \