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:
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:
|
||||
|
||||
Reference in New Issue
Block a user