Skip to content

Commit 0b88c06

Browse files
Update GPU allocation for vLLM container
1 parent 0b71acd commit 0b88c06

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

.github/workflows/test-p4d-4gpus.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ jobs:
1818

1919
- name: Start vLLM container (4 GPUs)
2020
run: |
21-
CONTAINER_ID=$(docker run -d --rm --gpus '"device=1,2"' \
21+
CONTAINER_ID=$(docker run -d --rm --gpus all \
2222
-p 8000:8000 \
2323
public.ecr.aws/deep-learning-containers/vllm:0.13.0-gpu-py312-ec2)
2424
echo "CONTAINER_ID=$CONTAINER_ID" >> $GITHUB_ENV

0 commit comments

Comments
 (0)