From 2d46ecaffd89499778c11276b0fe7d842f1d0a0e Mon Sep 17 00:00:00 2001 From: Paul Zhang Date: Thu, 5 Dec 2024 11:38:00 -0800 Subject: [PATCH] Update unit test CI to install torch using pip (#2604) Summary: Moving away from conda install for TorchRec CI, which atalman suggests Differential Revision: D66708069 --- .github/workflows/pyre.yml | 2 +- .github/workflows/unittest_ci.yml | 10 ++++++---- .github/workflows/unittest_ci_cpu.yml | 8 +++++--- 3 files changed, 12 insertions(+), 8 deletions(-) diff --git a/.github/workflows/pyre.yml b/.github/workflows/pyre.yml index be99ff62c..ed25404e7 100644 --- a/.github/workflows/pyre.yml +++ b/.github/workflows/pyre.yml @@ -19,7 +19,7 @@ jobs: uses: actions/checkout@v2 - name: Install dependencies run: > - conda install --yes pytorch cpuonly -c pytorch-nightly && + pip install torch --index-url https://download.pytorch.org/whl/nightly/cpu && pip install fbgemm-gpu --index-url https://download.pytorch.org/whl/nightly/cpu && pip install -r requirements.txt && pip install pyre-check-nightly==$(cat .pyre_configuration | grep version | awk '{print $2}' | sed 's/\"//g') diff --git a/.github/workflows/unittest_ci.yml b/.github/workflows/unittest_ci.yml index 8865acee4..0b3fe0638 100644 --- a/.github/workflows/unittest_ci.yml +++ b/.github/workflows/unittest_ci.yml @@ -73,13 +73,15 @@ 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/${{ matrix.cuda-tag }} + 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 \ - pip install fbgemm-gpu --index-url https://download.pytorch.org/whl/nightly/cu118 + pip install fbgemm-gpu --index-url https://download.pytorch.org/whl/nightly/${{ matrix.cuda-tag }} conda run -n build_binary \ python -c "import fbgemm_gpu" echo "fbgemm_gpu succeeded" diff --git a/.github/workflows/unittest_ci_cpu.yml b/.github/workflows/unittest_ci_cpu.yml index 2861029e1..1efe64178 100644 --- a/.github/workflows/unittest_ci_cpu.yml +++ b/.github/workflows/unittest_ci_cpu.yml @@ -45,9 +45,11 @@ jobs: conda info python --version conda run -n build_binary python --version - conda install -n build_binary \ - --yes \ - pytorch cpuonly -c pytorch-nightly + conda run -n build_binary \ + pip install torch --index-url https://download.pytorch.org/whl/nightly/cpu + 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 \