Skip to content

Commit

Permalink
Explicitly install torch cpu in SHARK CI
Browse files Browse the repository at this point in the history
  • Loading branch information
monorimet authored Jul 15, 2024
1 parent c95c645 commit 5bdb025
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions .github/workflows/test_shark.yml
Original file line number Diff line number Diff line change
Expand Up @@ -49,6 +49,7 @@ jobs:
cd $GITHUB_WORKSPACE/SHARK
python${{ matrix.version }} -m venv shark.venv
source shark.venv/bin/activate
pip install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cpu
pip install -r requirements.txt --no-cache-dir
pip install -e .
python apps/shark_studio/tests/api_test.py

0 comments on commit 5bdb025

Please sign in to comment.