mirror of
https://github.com/open-compass/opencompass.git
synced 2025-05-30 16:03:24 +08:00
update test torch version
This commit is contained in:
parent
3451c8190c
commit
0dcb1bd0fe
4
.github/workflows/pr-stage-check.yml
vendored
4
.github/workflows/pr-stage-check.yml
vendored
@ -20,7 +20,7 @@ jobs:
|
||||
matrix:
|
||||
python-version: ['3.10']
|
||||
include:
|
||||
- torch: 2.0.0
|
||||
- torch: 2.5.1
|
||||
steps:
|
||||
- uses: actions/checkout@v3
|
||||
- name: Set up Python ${{ matrix.python-version }}
|
||||
@ -30,7 +30,7 @@ jobs:
|
||||
- name: Upgrade pip
|
||||
run: python -m pip install --upgrade pip
|
||||
- name: Install PyTorch
|
||||
run: pip install torch==${{matrix.torch}}+cpu -f https://download.pytorch.org/whl/cpu/torch_stable.html
|
||||
run: pip install torch==${{matrix.torch}} -f https://download.pytorch.org/whl/cpu/torch_stable.html
|
||||
- name: Install system dependencies
|
||||
run: |
|
||||
sudo sed -i '$ a deb http://th.archive.ubuntu.com/ubuntu jammy main' /etc/apt/sources.list
|
||||
|
Loading…
Reference in New Issue
Block a user