github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-07-19 18:00:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

[ci] fixed compatibility workflow (#678)

Browse Source
This commit is contained in:
Frank Lee 2022-04-06 16:19:34 +08:00 committed by GitHub
parent 59bf2dc590
commit eace69387d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
.github/workflows/compatibility_test.yml vendored
View File

@ -6,14 +6,14 @@ jobs:
build: build:
name: Test for PyTorch compatibility name: Test for PyTorch compatibility
if: github.base_ref == 'main' && github.repository == 'hpcaitech/ColossalAI' if: github.base_ref == 'main' && github.repository == 'hpcaitech/ColossalAI'
runs-on: [self-hosted, aws] runs-on: [self-hosted, gpu]
strategy: strategy:
fail-fast: false fail-fast: false
matrix: matrix:
# 20.07: PyTorch 1.6.0a0+9907a3e + Python 3.6 # pytorch-cuda:1.9.0-11.1.1 + Python 3.8
# 20.10: PyTorch 1.7.0a0+7036e91 + Python 3.8 # pytorch-cuda:1.8.1-11.1.1 + Python 3.8
# 20.12: PyTorch 1.8.0a0+1606899 + Python 3.8 # pytorch-cuda:1.7.1-11.0.3 + Python 3.8
# 21.06: PyTorch 1.9.0a0+c3d40fd + Python 3.8 # pytorch-cuda:1.6.0-10.2 + Python 3.6
container: ["frankleeeee/pytorch-cuda:1.9.0-11.1.1", container: ["frankleeeee/pytorch-cuda:1.9.0-11.1.1",
"frankleeeee/pytorch-cuda:1.8.1-11.1.1", "frankleeeee/pytorch-cuda:1.8.1-11.1.1",
"frankleeeee/pytorch-cuda:1.7.1-11.0.3", "frankleeeee/pytorch-cuda:1.7.1-11.0.3",
@ -27,7 +27,7 @@ jobs:
run: | run: |
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
pip install -U pip setuptools wheel --user pip install -U pip setuptools wheel --user
pip install pytest tensorboard deepspeed apex pip install pytest tensorboard
- uses: actions/checkout@v2 - uses: actions/checkout@v2
- name: Install Colossal-AI - name: Install Colossal-AI
run: | run: |