SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

fix --gpus option for docker (#17235)

Browse Source 
Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar
2022-05-13 17:26:26 +02:00
committed by GitHub
parent 38043d8453
commit 993553b2f1
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
.github/workflows/self-scheduled.yml vendored
View File

@@ -126,7 +126,7 @@ jobs:
runs-on: ${{ matrix.machines }} runs-on: ${{ matrix.machines }}
container: container:
image: huggingface/transformers-all-latest-gpu 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 needs: setup
steps: steps:
- name: Echo folder ${{ matrix.folders }} - name: Echo folder ${{ matrix.folders }}
@@ -208,7 +208,7 @@ jobs:
runs-on: ${{ matrix.machines }} runs-on: ${{ matrix.machines }}
container: container:
image: huggingface/transformers-pytorch-gpu 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 needs: setup
steps: steps:
# Set machine type, i.e. `single-gpu` or `multi-gpu`. Here we just remove `-docker`. # Set machine type, i.e. `single-gpu` or `multi-gpu`. Here we just remove `-docker`.
@@ -252,7 +252,7 @@ jobs:
runs-on: ${{ matrix.machines }} runs-on: ${{ matrix.machines }}
container: container:
image: huggingface/transformers-tensorflow-gpu 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 needs: setup
steps: steps:
# Set machine type, i.e. `single-gpu` or `multi-gpu`. Here we just remove `-docker`. # Set machine type, i.e. `single-gpu` or `multi-gpu`. Here we just remove `-docker`.