diff --git a/.github/workflows/nightly_tests.yml b/.github/workflows/nightly_tests.yml index a863cfc11509..88a2af87c8f7 100644 --- a/.github/workflows/nightly_tests.yml +++ b/.github/workflows/nightly_tests.yml @@ -178,7 +178,7 @@ jobs: container: image: diffusers/diffusers-pytorch-cuda - options: --gpus 0 --shm-size "16gb" --ipc host + options: --gpus all --shm-size "16gb" --ipc host steps: - name: Checkout diffusers @@ -344,7 +344,7 @@ jobs: group: aws-g6e-xlarge-plus container: image: diffusers/diffusers-pytorch-cuda - options: --shm-size "20gb" --ipc host --gpus 0 + options: --shm-size "20gb" --ipc host --gpus all steps: - name: Checkout diffusers uses: actions/checkout@v3 @@ -396,7 +396,7 @@ jobs: group: aws-g6e-xlarge-plus container: image: diffusers/diffusers-pytorch-cuda - options: --shm-size "20gb" --ipc host --gpus 0 + options: --shm-size "20gb" --ipc host --gpus all steps: - name: Checkout diffusers uses: actions/checkout@v3 diff --git a/.github/workflows/pr_tests_gpu.yml b/.github/workflows/pr_tests_gpu.yml index 4179d9abf717..45294c89fe35 100644 --- a/.github/workflows/pr_tests_gpu.yml +++ b/.github/workflows/pr_tests_gpu.yml @@ -253,7 +253,7 @@ jobs: container: image: diffusers/diffusers-pytorch-cuda - options: --gpus 0 --shm-size "16gb" --ipc host + options: --gpus all --shm-size "16gb" --ipc host steps: - name: Checkout diffusers uses: actions/checkout@v3 diff --git a/.github/workflows/push_tests.yml b/.github/workflows/push_tests.yml index 499ef2467aca..6896e0145cbb 100644 --- a/.github/workflows/push_tests.yml +++ b/.github/workflows/push_tests.yml @@ -167,7 +167,7 @@ jobs: container: image: diffusers/diffusers-pytorch-cuda - options: --gpus 0 --shm-size "16gb" --ipc host + options: --gpus all --shm-size "16gb" --ipc host steps: - name: Checkout diffusers @@ -210,7 +210,7 @@ jobs: container: image: diffusers/diffusers-pytorch-xformers-cuda - options: --gpus 0 --shm-size "16gb" --ipc host + options: --gpus all --shm-size "16gb" --ipc host steps: - name: Checkout diffusers @@ -252,7 +252,7 @@ jobs: container: image: diffusers/diffusers-pytorch-cuda - options: --gpus 0 --shm-size "16gb" --ipc host + options: --gpus all --shm-size "16gb" --ipc host steps: - name: Checkout diffusers uses: actions/checkout@v3 diff --git a/.github/workflows/release_tests_fast.yml b/.github/workflows/release_tests_fast.yml index 75627a99c397..81a34f7a464d 100644 --- a/.github/workflows/release_tests_fast.yml +++ b/.github/workflows/release_tests_fast.yml @@ -222,7 +222,7 @@ jobs: container: image: diffusers/diffusers-pytorch-cuda - options: --gpus 0 --shm-size "16gb" --ipc host + options: --gpus all --shm-size "16gb" --ipc host steps: - name: Checkout diffusers @@ -265,7 +265,7 @@ jobs: container: image: diffusers/diffusers-pytorch-xformers-cuda - options: --gpus 0 --shm-size "16gb" --ipc host + options: --gpus all --shm-size "16gb" --ipc host steps: - name: Checkout diffusers @@ -307,7 +307,7 @@ jobs: container: image: diffusers/diffusers-pytorch-cuda - options: --gpus 0 --shm-size "16gb" --ipc host + options: --gpus all --shm-size "16gb" --ipc host steps: - name: Checkout diffusers diff --git a/.github/workflows/run_tests_from_a_pr.yml b/.github/workflows/run_tests_from_a_pr.yml index 94fbb2d297c5..c8eee8dbbc33 100644 --- a/.github/workflows/run_tests_from_a_pr.yml +++ b/.github/workflows/run_tests_from_a_pr.yml @@ -30,7 +30,7 @@ jobs: group: aws-g4dn-2xlarge container: image: ${{ github.event.inputs.docker_image }} - options: --gpus 0 --privileged --ipc host -v /mnt/cache/.cache/huggingface:/mnt/cache/ + options: --gpus all --privileged --ipc host -v /mnt/cache/.cache/huggingface:/mnt/cache/ steps: - name: Validate test files input