From 069d6fdc8462485e63269b20de92bf5c81d89ea7 Mon Sep 17 00:00:00 2001 From: Frank Lee Date: Mon, 18 Jul 2022 11:41:13 +0800 Subject: [PATCH] [workflow] update 8-gpu test to use torch 1.11 (#1332) --- .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 1e320be52..49003eeb5 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: