From 3c92fc62ca06b7bcc7c076fe27408b2313de168f Mon Sep 17 00:00:00 2001
From: shengguangming <shengguangming@bytedance.com>
Date: Wed, 8 Jan 2025 22:41:28 +0800
Subject: [PATCH] use docker on l20

---
 .github/workflows/gpu_test.yml | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/gpu_test.yml b/.github/workflows/gpu_test.yml
index 348d2412..65ff7790 100644
--- a/.github/workflows/gpu_test.yml
+++ b/.github/workflows/gpu_test.yml
@@ -19,18 +19,26 @@ on:
 jobs:
   ray:
     runs-on: [self-hosted, l20-0]
+    env:
+      HTTP_PROXY: ${{ secrets.PROXY_HTTP }}
+      HTTPS_PROXY: ${{ secrets.PROXY_HTTPS }}
+      NO_PROXY: "localhost,127.0.0.1"
+      HF_HUB_ENABLE_HF_TRANSFER: 1
+    container:
+      image: verlai/verl:vemlp-th2.4.0-cu124-vllm0.6.3-ray2.10-te1.7-v0.0.3
+      options: --gpus all --shm-size=10g
     steps:
       - uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4.2.2
         with:
             fetch-depth: 0
       - name: Install the current repository
         run: |
-          pip install -e .[test] --user
+          pip install -e .[test]
       - name: Running dataset tests
         run: |
           [ ! -d "$HOME/verl-data" ] && git clone --depth 1 https://github.com/eric-haibin-lin/verl-data ~/verl-data
           pytest -s -x tests/verl
-      - name: Running ray tests that need 2 GPUs
+      - name: Running ray tests that need 8 GPUs
         run: |
           cd tests/ray
           pytest -s -x .