Skip to content

GitLab

Unverified Commit fc84aff9 authored by Fengzhe Zhou's avatar Fengzhe Zhou Committed by GitHub
Browse files

[CI] Update github workflow cuda image (#874) 

* update workflow

* another trial

* another trial

* another trial
parent fed7d800
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
.github/workflows/pr-stage-check.yml
View file @ fc84aff9
...@@ -53,7 +53,7 @@ jobs: ...@@ -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']
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment