From 659a7407387754b70061ead5f98611a5bfeac060 Mon Sep 17 00:00:00 2001 From: Frank Lee Date: Fri, 15 Jul 2022 17:20:17 +0800 Subject: [PATCH] [workflow] roll back to use torch 1.11 for unit testing (#1325) --- .github/workflows/build.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml index fa00a6a72..25b89c6b7 100644 --- a/.github/workflows/build.yml +++ b/.github/workflows/build.yml @@ -14,7 +14,7 @@ jobs: contains( github.event.pull_request.labels.*.name, 'Run Build and Test') runs-on: [self-hosted, gpu] container: - image: hpcaitech/pytorch-cuda:1.12.0-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: