Skip to content

Commit

Permalink
use docker on l20
Browse files Browse the repository at this point in the history
  • Loading branch information
PeterSH6 committed Jan 8, 2025
1 parent c04fca8 commit 3c92fc6
Showing 1 changed file with 10 additions and 2 deletions.
12 changes: 10 additions & 2 deletions .github/workflows/gpu_test.yml
Original file line number Diff line number Diff line change
Expand Up @@ -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 .

0 comments on commit 3c92fc6

Please sign in to comment.