Skip to content

Commit

Permalink
fix: install cpu-only torch
Browse files Browse the repository at this point in the history
  • Loading branch information
yoyolicoris committed Mar 11, 2024
1 parent 4807222 commit 9f32bde
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 2 deletions.
3 changes: 2 additions & 1 deletion .github/workflows/pylint.yml
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,8 @@ jobs:
run: |
python -m pip install --upgrade pip
pip install pylint
if [ -f requirements.txt ]; then pip install -r requirements.txt; fi
pip install numpy numba
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
- name: Analysing the code with pylint
run: |
pylint $(git ls-files '*.py')
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 torchvision torchaudio --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 9f32bde

Please sign in to comment.