Skip to content

Commit 7b2513b

Browse files
nipung90facebook-github-bot
authored andcommitted
Fix torch version to make torchrec CI tests work (#3211)
Summary: Fix torch version to make torchrec CI tests work Rollback Plan: Differential Revision: D78564722
1 parent dd20e10 commit 7b2513b

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

.github/workflows/docs.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,7 @@ jobs:
5959
- name: Install PyTorch
6060
shell: bash
6161
run: |
62-
conda run -n build_binary pip install torch --index-url https://download.pytorch.org/whl/nightly/cpu
62+
conda run -n build_binary pip install torch==2.9.0.dev20250716+cpu --index-url https://download.pytorch.org/whl/nightly/cpu
6363
- name: Install fbgemm
6464
run: |
6565
conda run -n build_binary pip install fbgemm-gpu --index-url https://download.pytorch.org/whl/nightly/cpu

.github/workflows/unittest_ci.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -77,7 +77,7 @@ jobs:
7777
python --version
7878
conda run -n build_binary python --version
7979
conda run -n build_binary \
80-
pip install torch --index-url https://download.pytorch.org/whl/nightly/${{ matrix.cuda-tag }}
80+
pip install torch==2.9.0.dev20250716+${{ matrix.cuda-tag }} --index-url https://download.pytorch.org/whl/nightly/${{ matrix.cuda-tag }}
8181
conda run -n build_binary \
8282
python -c "import torch"
8383
echo "torch succeeded"

.github/workflows/unittest_ci_cpu.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,7 @@ jobs:
6161
python --version
6262
conda run -n build_binary python --version
6363
conda run -n build_binary \
64-
pip install torch --index-url https://download.pytorch.org/whl/nightly/cpu
64+
pip install torch==2.9.0.dev20250716+cpu --index-url https://download.pytorch.org/whl/nightly/cpu
6565
conda run -n build_binary \
6666
python -c "import torch"
6767
echo "torch succeeded"

0 commit comments

Comments
 (0)