diff --git a/.github/workflows/benchmark.yml b/.github/workflows/benchmark.yml index 747e1d815406..cc97e043c139 100644 --- a/.github/workflows/benchmark.yml +++ b/.github/workflows/benchmark.yml @@ -25,7 +25,7 @@ jobs: group: aws-g6e-4xlarge container: image: diffusers/diffusers-pytorch-cuda - options: --shm-size "16gb" --ipc host --gpus 0 + options: --shm-size "16gb" --ipc host --gpus all steps: - name: Checkout diffusers uses: actions/checkout@v3 diff --git a/.github/workflows/nightly_tests.yml b/.github/workflows/nightly_tests.yml index 384f07506afe..a863cfc11509 100644 --- a/.github/workflows/nightly_tests.yml +++ b/.github/workflows/nightly_tests.yml @@ -61,7 +61,7 @@ jobs: group: aws-g4dn-2xlarge container: image: diffusers/diffusers-pytorch-cuda - options: --shm-size "16gb" --ipc host --gpus 0 + options: --shm-size "16gb" --ipc host --gpus all steps: - name: Checkout diffusers uses: actions/checkout@v3 @@ -107,7 +107,7 @@ jobs: group: aws-g4dn-2xlarge container: image: diffusers/diffusers-pytorch-cuda - options: --shm-size "16gb" --ipc host --gpus 0 + options: --shm-size "16gb" --ipc host --gpus all defaults: run: shell: bash @@ -222,7 +222,7 @@ jobs: group: aws-g6e-xlarge-plus container: image: diffusers/diffusers-pytorch-cuda - options: --shm-size "16gb" --ipc host --gpus 0 + options: --shm-size "16gb" --ipc host --gpus all steps: - name: Checkout diffusers uses: actions/checkout@v3 @@ -270,7 +270,7 @@ jobs: group: aws-g4dn-2xlarge container: image: diffusers/diffusers-pytorch-minimum-cuda - options: --shm-size "16gb" --ipc host --gpus 0 + options: --shm-size "16gb" --ipc host --gpus all defaults: run: shell: bash diff --git a/.github/workflows/pr_tests_gpu.yml b/.github/workflows/pr_tests_gpu.yml index bb74daad214e..4179d9abf717 100644 --- a/.github/workflows/pr_tests_gpu.yml +++ b/.github/workflows/pr_tests_gpu.yml @@ -118,7 +118,7 @@ jobs: group: aws-g4dn-2xlarge container: image: diffusers/diffusers-pytorch-cuda - options: --shm-size "16gb" --ipc host --gpus 0 + options: --shm-size "16gb" --ipc host --gpus all steps: - name: Checkout diffusers uses: actions/checkout@v3 @@ -183,7 +183,7 @@ jobs: group: aws-g4dn-2xlarge container: image: diffusers/diffusers-pytorch-cuda - options: --shm-size "16gb" --ipc host --gpus 0 + options: --shm-size "16gb" --ipc host --gpus all defaults: run: shell: bash diff --git a/.github/workflows/push_tests.yml b/.github/workflows/push_tests.yml index 007770c8ed67..499ef2467aca 100644 --- a/.github/workflows/push_tests.yml +++ b/.github/workflows/push_tests.yml @@ -64,7 +64,7 @@ jobs: group: aws-g4dn-2xlarge container: image: diffusers/diffusers-pytorch-cuda - options: --shm-size "16gb" --ipc host --gpus 0 + options: --shm-size "16gb" --ipc host --gpus all steps: - name: Checkout diffusers uses: actions/checkout@v3 @@ -109,7 +109,7 @@ jobs: group: aws-g4dn-2xlarge container: image: diffusers/diffusers-pytorch-cuda - options: --shm-size "16gb" --ipc host --gpus 0 + options: --shm-size "16gb" --ipc host --gpus all defaults: run: shell: bash diff --git a/.github/workflows/release_tests_fast.yml b/.github/workflows/release_tests_fast.yml index e5d328204943..75627a99c397 100644 --- a/.github/workflows/release_tests_fast.yml +++ b/.github/workflows/release_tests_fast.yml @@ -62,7 +62,7 @@ jobs: group: aws-g4dn-2xlarge container: image: diffusers/diffusers-pytorch-cuda - options: --shm-size "16gb" --ipc host --gpus 0 + options: --shm-size "16gb" --ipc host --gpus all steps: - name: Checkout diffusers uses: actions/checkout@v3 @@ -107,7 +107,7 @@ jobs: group: aws-g4dn-2xlarge container: image: diffusers/diffusers-pytorch-cuda - options: --shm-size "16gb" --ipc host --gpus 0 + options: --shm-size "16gb" --ipc host --gpus all defaults: run: shell: bash @@ -163,7 +163,7 @@ jobs: group: aws-g4dn-2xlarge container: image: diffusers/diffusers-pytorch-minimum-cuda - options: --shm-size "16gb" --ipc host --gpus 0 + options: --shm-size "16gb" --ipc host --gpus all defaults: run: shell: bash