Skip to content

Commit

Permalink
Pin torch to 2.4.1
Browse files Browse the repository at this point in the history
  • Loading branch information
eagarvey-amd committed Oct 2, 2024
1 parent 0291d43 commit e337f2a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion .github/workflows/test_models.yml
Original file line number Diff line number Diff line change
Expand Up @@ -57,7 +57,7 @@ jobs:
# Note: We install in three steps in order to satisfy requirements
# from non default locations first. Installing the PyTorch CPU
# wheels saves multiple minutes and a lot of bandwidth on runner setup.
pip install --pre torch==2.5.0.dev20240801+cpu torchvision==0.20.0.dev20240801+cpu --index-url https://download.pytorch.org/whl/nightly/cpu
pip install --pre torch==2.4.1+cpu torchvision --index-url https://download.pytorch.org/whl/cpu
pip install --pre --upgrade -r ${{ github.workspace }}/iree-turbine/requirements.txt
pip install --no-compile --pre -e ${{ github.workspace }}/iree-turbine[testing]
pip install --upgrade --pre --no-cache-dir iree-compiler iree-runtime -f https://iree.dev/pip-release-links.html
Expand Down

0 comments on commit e337f2a

Please sign in to comment.