File tree Expand file tree Collapse file tree 3 files changed +3
-3
lines changed Expand file tree Collapse file tree 3 files changed +3
-3
lines changed Original file line number Diff line number Diff line change 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)
Original file line number Diff line number Diff line change 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 :
Original file line number Diff line number Diff line change 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
You can’t perform that action at this time.
0 commit comments