Skip to content

GitLab

Unverified Commit 659a7407 authored by Frank Lee's avatar Frank Lee Committed by GitHub
Browse files

[workflow] roll back to use torch 1.11 for unit testing (#1325)

parent 4d5dbf48
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
.github/workflows/build.yml
View file @ 659a7407
...@@ -14,7 +14,7 @@ jobs: ...@@ -14,7 +14,7 @@ jobs:
contains( github.event.pull_request.labels.*.name, 'Run Build and Test') contains( github.event.pull_request.labels.*.name, 'Run Build and Test')
runs-on: [self-hosted, gpu] runs-on: [self-hosted, gpu]
container: 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 options: --gpus all --rm -v /data/scratch/cifar-10:/data/scratch/cifar-10
timeout-minutes: 40 timeout-minutes: 40
steps: steps:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment