Skip to content

Commit

Permalink
Update wheels-generator.yml
Browse files Browse the repository at this point in the history
  • Loading branch information
OutofAi authored Feb 18, 2025
1 parent 1b1402e commit 4c31739
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions .github/workflows/wheels-generator.yml
Original file line number Diff line number Diff line change
Expand Up @@ -224,14 +224,14 @@ jobs:
ls -R
- name: Install Wheel
run:
pip install setuptools wheel
pip3.11 install setuptools wheel
- name: Install Pytorch
run: |
pip3 install torch==2.5.1 --index-url https://download.pytorch.org/whl/cu${{ matrix.cudaversion }}
pip3.11 install torch==2.5.1 --index-url https://download.pytorch.org/whl/cu${{ matrix.cudaversion }}
- name: Generate Wheels
run: |
cd bindings/torch
python3 setup.py bdist_wheel --plat-name linux_x86_64
python3.11 setup.py bdist_wheel --plat-name linux_x86_64
env:
INCLUDE_ARCH: ${{ matrix.arch }}
- name: Upload assets
Expand Down

0 comments on commit 4c31739

Please sign in to comment.