From b15d0fe28771fef0ae95f0e8064f106ad47e0f4a Mon Sep 17 00:00:00 2001
From: ShawnXuan <xiexuanx2@gmail.com>
Date: Thu, 9 Nov 2023 12:47:13 +0800
Subject: [PATCH] update

---
 .github/workflows/test.yml | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index 88ce6afc..91e67254 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -16,6 +16,7 @@ concurrency:
 env:
   CO_MODELS_SRC: CoModels
   TEST_DIR: test_dir
+  TEST_CONTAINER_NAME: registry.cn-beijing.aliyuncs.com/oneflow/manylinux2014_x86_64_cuda12.2
 
 jobs:
   test_job:
@@ -42,13 +43,13 @@ jobs:
           path: ${{ env.CO_MODELS_SRC}}
       - name: Prepare test directory
         run: |
-          docker run -it --rm --runtime=nvidia --privileged \
-            --network host --gpus=all \
-            --ipc=host \
+          docker run --gpus=all -d --rm --privileged --shm-size=8g \
+            --pids-limit 2000 \
+            --cap-add=SYS_PTRACE --security-opt seccomp=unconfined \
             -v /share_nfs:/data \
             -w /data \
-            registry.cn-beijing.aliyuncs.com/oneflow/manylinux2014_x86_64_cuda12.2 \
-            ls
+            --name ${TEST_CONTAINER_NAME} \
+            ls && python3 -m oneflow --doctor 
           #model_dir='${{ env.CO_MODELS_SRC }}/cv/classification/${{ env.BRANCH }}'
           #docker images | grep oneflow
           #model_dir='${{ env.CO_MODELS_SRC }}/cv/classification/resnet50'