Skip to content

Commit 7e05056

Browse files
: GHA: pin torch-2.9.0 (#1742)
Summary: Pull Request resolved: #1742 Differential Revision: D86052365
1 parent f40f6db commit 7e05056

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

.github/workflows/build-cpu.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ jobs:
3030
setup_build_environment
3131
3232
# Install torch nightly (CPU version)
33-
pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu
33+
pip install torch==2.9.0+cpu --index-url https://download.pytorch.org/whl/cpu
3434
pip install -r build-requirements.txt
3535
3636
# Build monarch (No tensor engine, CPU version)

.github/workflows/build-cuda.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ jobs:
1717
include:
1818
- name: 4xlargegpu
1919
runs-on: linux.g5.4xlarge.nvidia.gpu
20-
torch-spec: '--pre torch --index-url https://download.pytorch.org/whl/nightly/cu126'
20+
torch-spec: 'torch==2.9.0+cu126 --index-url https://download.pytorch.org/whl/cu126'
2121
gpu-arch-type: "cuda"
2222
gpu-arch-version: "12.6"
2323
with:

.github/workflows/doc_build.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@ jobs:
3939
pip install -r docs/requirements.txt
4040
4141
# Install PyTorch with CUDA support (matching build-cuda.yml)
42-
pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu126
42+
pip install torch==2.9.0+cu126 --index-url https://download.pytorch.org/whl/cu126
4343
4444
# Set environment variables for CUDA build
4545
export USE_CUDA=1

0 commit comments

Comments
 (0)