Skip to content

Commit ff4725e

Browse files
committed
fix command
Signed-off-by: Junpu Fan <junpu@amazon.com>
1 parent 65975f7 commit ff4725e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

.github/workflows/pr-example.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -87,4 +87,4 @@ jobs:
8787
run: |
8888
aws ecr get-login-password --region ${{ secrets.AWS_REGION }} | docker login --username AWS --password-stdin ${{ secrets.AWS_ACCOUNT_ID }}.dkr.ecr.${{ secrets.AWS_REGION }}.amazonaws.com
8989
docker pull "$IMAGE_URI"
90-
docker run --rm "$IMAGE_URI" --entrypoint python -c "import torch; import ray; import vllm; print('Test OK')"
90+
docker run --rm --gpus=all --entrypoint /bin/bash "$IMAGE_URI" --entrypoint python -c "import torch; import ray; import vllm; print('Test OK')"

0 commit comments

Comments
 (0)