diff --git a/.github/workflows/benchmark.yml b/.github/workflows/benchmark.yml index 747e1d8154..cc97e043c1 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 384f07506a..a863cfc115 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 bb74daad21..4179d9abf7 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 007770c8ed..499ef2467a 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 e5d3282049..75627a99c3 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