From fc84aff963bf9bbc6baf903f2ff23af9f4c5534c Mon Sep 17 00:00:00 2001 From: Fengzhe Zhou Date: Mon, 5 Feb 2024 15:22:59 +0800 Subject: [PATCH] [CI] Update github workflow cuda image (#874) * update workflow * another trial * another trial * another trial --- .github/workflows/pr-stage-check.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/pr-stage-check.yml b/.github/workflows/pr-stage-check.yml index ca0b9f1e..8d2c62bd 100644 --- a/.github/workflows/pr-stage-check.yml +++ b/.github/workflows/pr-stage-check.yml @@ -53,7 +53,7 @@ jobs: build_cu117: runs-on: ubuntu-22.04 container: - image: pytorch/pytorch:2.0.0-cuda11.7-cudnn8-devel + image: nvidia/cuda:11.7.1-cudnn8-runtime-ubuntu22.04 strategy: matrix: python-version: ['3.10']