Update setup_github_runner.md (#1952)
This commit is contained in:
@@ -1,4 +1,4 @@
|
|||||||
# Set Up Self-hosted Runners for GitHub Action
|
# Set Up Self-Hosted Runners for GitHub Action
|
||||||
|
|
||||||
## Add a Runner
|
## Add a Runner
|
||||||
|
|
||||||
@@ -9,9 +9,9 @@ You can mount a folder for the shared huggingface model weights cache. The comma
|
|||||||
```
|
```
|
||||||
docker pull nvidia/cuda:12.1.1-devel-ubuntu22.04
|
docker pull nvidia/cuda:12.1.1-devel-ubuntu22.04
|
||||||
# Nvidia
|
# Nvidia
|
||||||
docker run --shm-size 64g -it -v /tmp/huggingface:/hf_home --gpus all nvidia/cuda:12.1.1-devel-ubuntu22.04 /bin/bash
|
docker run --shm-size 128g -it -v /tmp/huggingface:/hf_home --gpus all nvidia/cuda:12.1.1-devel-ubuntu22.04 /bin/bash
|
||||||
# AMD
|
# AMD
|
||||||
docker run --rm --device=/dev/kfd --device=/dev/dri --group-add video --shm-size 64g -it -v /tmp/huggingface:/hf_home henryx/haisgl:sgl0.3.1.post3_vllm0.6.0_triton3.0.0_rocm6.2.1 /bin/bash
|
docker run --rm --device=/dev/kfd --device=/dev/dri --group-add video --shm-size 128g -it -v /tmp/huggingface:/hf_home henryx/haisgl:sgl0.3.1.post3_vllm0.6.0_triton3.0.0_rocm6.2.1 /bin/bash
|
||||||
```
|
```
|
||||||
|
|
||||||
### Step 2: Configure the runner by `config.sh`
|
### Step 2: Configure the runner by `config.sh`
|
||||||
|
|||||||
Reference in New Issue
Block a user