diff --git a/.github/workflows/docs.yml b/.github/workflows/docs.yml index 547fd0d77..a24c634c3 100644 --- a/.github/workflows/docs.yml +++ b/.github/workflows/docs.yml @@ -59,7 +59,7 @@ jobs: - name: Install PyTorch shell: bash run: | - conda run -n build_binary pip install torch --index-url https://download.pytorch.org/whl/nightly/cpu + conda run -n build_binary pip install torch==2.9.0.dev20250716+cpu --index-url https://download.pytorch.org/whl/nightly/cpu - name: Install fbgemm run: | conda run -n build_binary pip install fbgemm-gpu --index-url https://download.pytorch.org/whl/nightly/cpu diff --git a/.github/workflows/unittest_ci.yml b/.github/workflows/unittest_ci.yml index 5fdce0115..2be5908a0 100644 --- a/.github/workflows/unittest_ci.yml +++ b/.github/workflows/unittest_ci.yml @@ -77,7 +77,7 @@ jobs: python --version conda run -n build_binary python --version conda run -n build_binary \ - pip install torch --index-url https://download.pytorch.org/whl/nightly/${{ matrix.cuda-tag }} + pip install torch==2.9.0.dev20250716+${{ matrix.cuda-tag }} --index-url https://download.pytorch.org/whl/nightly/${{ matrix.cuda-tag }} conda run -n build_binary \ python -c "import torch" echo "torch succeeded" diff --git a/.github/workflows/unittest_ci_cpu.yml b/.github/workflows/unittest_ci_cpu.yml index 7289fa509..ad3fec7a9 100644 --- a/.github/workflows/unittest_ci_cpu.yml +++ b/.github/workflows/unittest_ci_cpu.yml @@ -61,7 +61,7 @@ jobs: python --version conda run -n build_binary python --version conda run -n build_binary \ - pip install torch --index-url https://download.pytorch.org/whl/nightly/cpu + pip install torch==2.9.0.dev20250716+cpu --index-url https://download.pytorch.org/whl/nightly/cpu conda run -n build_binary \ python -c "import torch" echo "torch succeeded"