mirror of
https://github.com/huggingface/transformers.git
synced 2025-07-03 21:00:08 +06:00
At most 2 GPUs for CI (#27435)
At most 2 GPUs Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
parent
68afca3e69
commit
00dc856233
1
.github/workflows/self-nightly-scheduled.yml
vendored
1
.github/workflows/self-nightly-scheduled.yml
vendored
@ -19,6 +19,7 @@ env:
|
||||
SIGOPT_API_TOKEN: ${{ secrets.SIGOPT_API_TOKEN }}
|
||||
TF_FORCE_GPU_ALLOW_GROWTH: true
|
||||
RUN_PT_TF_CROSS_TESTS: 1
|
||||
CUDA_VISIBLE_DEVICES: 0,1
|
||||
|
||||
jobs:
|
||||
setup:
|
||||
|
1
.github/workflows/self-past.yml
vendored
1
.github/workflows/self-past.yml
vendored
@ -30,6 +30,7 @@ env:
|
||||
SIGOPT_API_TOKEN: ${{ secrets.SIGOPT_API_TOKEN }}
|
||||
TF_FORCE_GPU_ALLOW_GROWTH: true
|
||||
RUN_PT_TF_CROSS_TESTS: 1
|
||||
CUDA_VISIBLE_DEVICES: 0,1
|
||||
|
||||
jobs:
|
||||
setup:
|
||||
|
1
.github/workflows/self-push.yml
vendored
1
.github/workflows/self-push.yml
vendored
@ -25,6 +25,7 @@ env:
|
||||
PYTEST_TIMEOUT: 60
|
||||
TF_FORCE_GPU_ALLOW_GROWTH: true
|
||||
RUN_PT_TF_CROSS_TESTS: 1
|
||||
CUDA_VISIBLE_DEVICES: 0,1
|
||||
|
||||
jobs:
|
||||
setup:
|
||||
|
1
.github/workflows/self-scheduled.yml
vendored
1
.github/workflows/self-scheduled.yml
vendored
@ -23,6 +23,7 @@ env:
|
||||
SIGOPT_API_TOKEN: ${{ secrets.SIGOPT_API_TOKEN }}
|
||||
TF_FORCE_GPU_ALLOW_GROWTH: true
|
||||
RUN_PT_TF_CROSS_TESTS: 1
|
||||
CUDA_VISIBLE_DEVICES: 0,1
|
||||
|
||||
jobs:
|
||||
setup:
|
||||
|
Loading…
Reference in New Issue
Block a user