diff --git a/.github/workflows/unittest_ci.yml b/.github/workflows/unittest_ci.yml index 2f5d8a2dd..1c8cae063 100644 --- a/.github/workflows/unittest_ci.yml +++ b/.github/workflows/unittest_ci.yml @@ -82,7 +82,7 @@ jobs: 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/${{ matrix.cuda-tag }} + pip install --pre 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 7289fa509..518acd050 100644 --- a/.github/workflows/unittest_ci_cpu.yml +++ b/.github/workflows/unittest_ci_cpu.yml @@ -68,7 +68,7 @@ jobs: 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/cpu + pip install --pre fbgemm-gpu --index-url https://download.pytorch.org/whl/nightly/cpu conda run -n build_binary \ python -c "import fbgemm_gpu" echo "fbgemm_gpu succeeded"