github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-08-21 01:13:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Frank Lee 2022-07-18 11:41:13 +08:00 committed by GitHub
parent 7857fd7616
commit 069d6fdc84
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.github/workflows/build_gpu_8.yml vendored
View File

@ -12,7 +12,7 @@ jobs:
if: github.repository == 'hpcaitech/ColossalAI' if: github.repository == 'hpcaitech/ColossalAI'
runs-on: [self-hosted, 8-gpu] runs-on: [self-hosted, 8-gpu]
container: 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 options: --gpus all --rm -v /data/scratch/cifar-10:/data/scratch/cifar-10
timeout-minutes: 40 timeout-minutes: 40
steps: steps: