From 65b9e0ece5c6e2dcac141717f27b8be7d03f0d39 Mon Sep 17 00:00:00 2001 From: Darrel Pol Date: Sat, 2 Nov 2024 16:28:43 -0500 Subject: [PATCH] adding gpus all flag --- .github/workflows/on_push_deploy_to_staging.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/on_push_deploy_to_staging.yaml b/.github/workflows/on_push_deploy_to_staging.yaml index db0acf3e5..1e02583fc 100644 --- a/.github/workflows/on_push_deploy_to_staging.yaml +++ b/.github/workflows/on_push_deploy_to_staging.yaml @@ -75,7 +75,7 @@ jobs: docker rm moonlite-sd || true # Run the new Docker container - docker run -d -p 7861:7861 --name moonlite-sd \ + docker run --gpus all -d -p 7861:7861 --name moonlite-sd \ --env-file /home/${{ secrets.STAGING_EC2_USER }}/.env \ ${{ secrets.STAGING_ECR_URI }}:latest