From c36ae1b049f41cfc1f3f90a136c8dafc05d1d3c2 Mon Sep 17 00:00:00 2001 From: FrankLeeeee Date: Mon, 18 Jul 2022 11:39:11 +0800 Subject: [PATCH] [workflow] update 8-gpu test to use torch 1.11 --- .github/workflows/build_gpu_8.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/build_gpu_8.yml b/.github/workflows/build_gpu_8.yml index 1e320be52465..49003eeb5bed 100644 --- a/.github/workflows/build_gpu_8.yml +++ b/.github/workflows/build_gpu_8.yml @@ -12,7 +12,7 @@ jobs: if: github.repository == 'hpcaitech/ColossalAI' runs-on: [self-hosted, 8-gpu] container: - image: hpcaitech/pytorch-cuda:1.10.1-11.3.0 + image: hpcaitech/pytorch-cuda:1.11.0-11.3.0 options: --gpus all --rm -v /data/scratch/cifar-10:/data/scratch/cifar-10 timeout-minutes: 40 steps: