Skip to content

GitLab

Unverified Commit 993553b2 authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

fix --gpus option for docker (#17235) 


Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent 38043d84
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
.github/workflows/self-scheduled.yml
View file @ 993553b2
......@@ -126,7 +126,7 @@ jobs:
runs-on: ${{ matrix.machines }}
container:
image: huggingface/transformers-all-latest-gpu
options: --gpus 0 --shm-size "16gb" --ipc host -v /mnt/cache/.cache/huggingface:/mnt/cache/
options: --gpus all --shm-size "16gb" --ipc host -v /mnt/cache/.cache/huggingface:/mnt/cache/
needs: setup
steps:
- name: Echo folder ${{ matrix.folders }}
......@@ -208,7 +208,7 @@ jobs:
runs-on: ${{ matrix.machines }}
container:
image: huggingface/transformers-pytorch-gpu
options: --gpus 0 --shm-size "16gb" --ipc host -v /mnt/cache/.cache/huggingface:/mnt/cache/
options: --gpus all --shm-size "16gb" --ipc host -v /mnt/cache/.cache/huggingface:/mnt/cache/
needs: setup
steps:
# Set machine type, i.e. `single-gpu` or `multi-gpu`. Here we just remove `-docker`.
......@@ -252,7 +252,7 @@ jobs:
runs-on: ${{ matrix.machines }}
container:
image: huggingface/transformers-tensorflow-gpu
options: --gpus 0 --shm-size "16gb" --ipc host -v /mnt/cache/.cache/huggingface:/mnt/cache/
options: --gpus all --shm-size "16gb" --ipc host -v /mnt/cache/.cache/huggingface:/mnt/cache/
needs: setup
steps:
# Set machine type, i.e. `single-gpu` or `multi-gpu`. Here we just remove `-docker`.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment