mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2026-01-13 19:46:01 +00:00
44 lines
1.4 KiB
YAML
44 lines
1.4 KiB
YAML
name: Compatibility Test
|
|
|
|
on: workflow_dispatch
|
|
|
|
jobs:
|
|
build:
|
|
name: Test for PyTorch compatibility
|
|
if: github.base_ref == 'main' && github.repository == 'hpcaitech/ColossalAI'
|
|
runs-on: [self-hosted, aws]
|
|
strategy:
|
|
fail-fast: false
|
|
matrix:
|
|
# 20.07: PyTorch 1.6.0a0+9907a3e + Python 3.6
|
|
# 20.10: PyTorch 1.7.0a0+7036e91 + Python 3.8
|
|
# 20.12: PyTorch 1.8.0a0+1606899 + Python 3.8
|
|
# 21.06: PyTorch 1.9.0a0+c3d40fd + Python 3.8
|
|
container: ["frankleeeee/pytorch-cuda:1.9.0-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.6.0-10.2"]
|
|
container:
|
|
image: ${{ matrix.container }}
|
|
options: --gpus all --rm --ipc=host -v /data/scratch/cifar-10:/data/scratch/cifar-10
|
|
timeout-minutes: 120
|
|
steps:
|
|
- name: Install dependencies
|
|
run: |
|
|
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
|
|
pip install -U pip setuptools wheel --user
|
|
pip install pytest tensorboard deepspeed apex
|
|
- uses: actions/checkout@v2
|
|
- name: Install Colossal-AI
|
|
run: |
|
|
pip install -r requirements/requirements.txt
|
|
pip install -v --no-cache-dir .
|
|
- name: Unit Testing
|
|
run: |
|
|
PYTHONPATH=$PWD pytest tests
|
|
env:
|
|
DATA: /data/scratch/cifar-10
|
|
|
|
|
|
|