Skip to content

GitLab

Unverified Commit 00dc8562 authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

At most 2 GPUs for CI (#27435) 



At most 2 GPUs
Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent 68afca3e
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
.github/workflows/self-nightly-scheduled.yml
View file @ 00dc8562
...@@ -19,6 +19,7 @@ env: ...@@ -19,6 +19,7 @@ env:
SIGOPT_API_TOKEN: ${{ secrets.SIGOPT_API_TOKEN }} SIGOPT_API_TOKEN: ${{ secrets.SIGOPT_API_TOKEN }}
TF_FORCE_GPU_ALLOW_GROWTH: true TF_FORCE_GPU_ALLOW_GROWTH: true
RUN_PT_TF_CROSS_TESTS: 1 RUN_PT_TF_CROSS_TESTS: 1
CUDA_VISIBLE_DEVICES: 0,1
jobs: jobs:
setup: setup:
......
.github/workflows/self-past.yml
View file @ 00dc8562
...@@ -30,6 +30,7 @@ env: ...@@ -30,6 +30,7 @@ env:
SIGOPT_API_TOKEN: ${{ secrets.SIGOPT_API_TOKEN }} SIGOPT_API_TOKEN: ${{ secrets.SIGOPT_API_TOKEN }}
TF_FORCE_GPU_ALLOW_GROWTH: true TF_FORCE_GPU_ALLOW_GROWTH: true
RUN_PT_TF_CROSS_TESTS: 1 RUN_PT_TF_CROSS_TESTS: 1
CUDA_VISIBLE_DEVICES: 0,1
jobs: jobs:
setup: setup:
......
.github/workflows/self-push.yml
View file @ 00dc8562
...@@ -25,6 +25,7 @@ env: ...@@ -25,6 +25,7 @@ env:
PYTEST_TIMEOUT: 60 PYTEST_TIMEOUT: 60
TF_FORCE_GPU_ALLOW_GROWTH: true TF_FORCE_GPU_ALLOW_GROWTH: true
RUN_PT_TF_CROSS_TESTS: 1 RUN_PT_TF_CROSS_TESTS: 1
CUDA_VISIBLE_DEVICES: 0,1
jobs: jobs:
setup: setup:
......
.github/workflows/self-scheduled.yml
View file @ 00dc8562
...@@ -23,6 +23,7 @@ env: ...@@ -23,6 +23,7 @@ env:
SIGOPT_API_TOKEN: ${{ secrets.SIGOPT_API_TOKEN }} SIGOPT_API_TOKEN: ${{ secrets.SIGOPT_API_TOKEN }}
TF_FORCE_GPU_ALLOW_GROWTH: true TF_FORCE_GPU_ALLOW_GROWTH: true
RUN_PT_TF_CROSS_TESTS: 1 RUN_PT_TF_CROSS_TESTS: 1
CUDA_VISIBLE_DEVICES: 0,1
jobs: jobs:
setup: setup:
......
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