[workflow] update 8-gpu test to use torch 1.11 (#1332)

069d6fdc · Frank Lee · GitHub · 7857fd76 · 069d6fdc
Unverified Commit 069d6fdc authored Jul 18, 2022 by Frank Lee Committed by GitHub Jul 18, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

.github/workflows/build_gpu_8.yml .github/workflows/build_gpu_8.yml +1 -1

No files found.
--- 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: