Unverified Commit 4cf2f12b authored by Przemyslaw Tredak's avatar Przemyslaw Tredak Committed by GitHub
Browse files

Change the pyTorch installation to CUDA 13 in Build All GitHub action (#2308)



Change the pyTorch installation to CUDA 13 in Build All GitHub action to
match the version in the JAX container
Signed-off-by: default avatarPrzemek Tredak <ptredak@nvidia.com>
parent a019c80a
...@@ -83,7 +83,9 @@ jobs: ...@@ -83,7 +83,9 @@ jobs:
options: --user root options: --user root
steps: steps:
- name: 'Dependencies' - name: 'Dependencies'
run: pip install torch pybind11[global] einops onnxscript run: |
pip install pybind11[global] einops onnxscript
pip install torch --index-url https://download.pytorch.org/whl/cu130
- name: 'Checkout' - name: 'Checkout'
uses: actions/checkout@v3 uses: actions/checkout@v3
with: with:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment