Skip to content

Commit

Permalink
use cpu-only version torch for workflows
Browse files Browse the repository at this point in the history
  • Loading branch information
yoyolicoris committed Apr 18, 2024
1 parent edad7aa commit 50547b5
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion .github/workflows/python-package.yml
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,8 @@ jobs:
run: |
python -m pip install --upgrade pip
python -m pip install flake8 pytest
if [ -f requirements.txt ]; then pip install -r requirements.txt; fi
pip install numpy numba
pip install torch --index-url https://download.pytorch.org/whl/cpu
- name: Lint with flake8
run: |
# stop the build if there are Python syntax errors or undefined names
Expand Down

0 comments on commit 50547b5

Please sign in to comment.