mirror of
https://github.com/open-compass/opencompass.git
synced 2025-05-30 16:03:24 +08:00
[CI] Update github workflow cuda image (#874)
* update workflow * another trial * another trial * another trial
This commit is contained in:
parent
fed7d800c6
commit
fc84aff963
2
.github/workflows/pr-stage-check.yml
vendored
2
.github/workflows/pr-stage-check.yml
vendored
@ -53,7 +53,7 @@ jobs:
|
|||||||
build_cu117:
|
build_cu117:
|
||||||
runs-on: ubuntu-22.04
|
runs-on: ubuntu-22.04
|
||||||
container:
|
container:
|
||||||
image: pytorch/pytorch:2.0.0-cuda11.7-cudnn8-devel
|
image: nvidia/cuda:11.7.1-cudnn8-runtime-ubuntu22.04
|
||||||
strategy:
|
strategy:
|
||||||
matrix:
|
matrix:
|
||||||
python-version: ['3.10']
|
python-version: ['3.10']
|
||||||
|
Loading…
Reference in New Issue
Block a user