[devops] fix example test ci (#5504)

This commit is contained in:
Hongxin Liu
2024-03-26 15:09:05 +08:00
committed by GitHub
parent 131f32a076
commit a7790a92e8
3 changed files with 3 additions and 8 deletions

View File

@@ -35,6 +35,7 @@ jobs:
matrix: ${{fromJson(needs.matrix_preparation.outputs.matrix)}}
container:
image: hpcaitech/pytorch-cuda:2.1.0-12.1.0
options: --gpus all --rm -v /data/scratch/examples-data:/data/ -v /dev/shm
timeout-minutes: 10
steps:
- name: 📚 Checkout
@@ -50,8 +51,6 @@ jobs:
echo "Testing ${example_dir} now"
cd "${PWD}/examples/${example_dir}"
bash test_ci.sh
env:
NCCL_SHM_DISABLE: 1
- name: Notify Lark
id: message-preparation