We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 65975f7 commit ff4725eCopy full SHA for ff4725e
.github/workflows/pr-example.yml
@@ -87,4 +87,4 @@ jobs:
87
run: |
88
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
89
docker pull "$IMAGE_URI"
90
- docker run --rm "$IMAGE_URI" --entrypoint python -c "import torch; import ray; import vllm; print('Test OK')"
+ 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