From 50547b54d7fa25a49b30ebb19b2256e37e3336fa Mon Sep 17 00:00:00 2001 From: Chin-Yun Yu Date: Fri, 19 Apr 2024 02:11:27 +0800 Subject: [PATCH] use cpu-only version torch for workflows --- .github/workflows/python-package.yml | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/.github/workflows/python-package.yml b/.github/workflows/python-package.yml index f6f35d7..03fbf19 100644 --- a/.github/workflows/python-package.yml +++ b/.github/workflows/python-package.yml @@ -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