diff --git a/.github/workflows/self-nightly-scheduled.yml b/.github/workflows/self-nightly-scheduled.yml index f5d96828183..07a6197584a 100644 --- a/.github/workflows/self-nightly-scheduled.yml +++ b/.github/workflows/self-nightly-scheduled.yml @@ -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: diff --git a/.github/workflows/self-past.yml b/.github/workflows/self-past.yml index ad624fb2581..568b6a9b64f 100644 --- a/.github/workflows/self-past.yml +++ b/.github/workflows/self-past.yml @@ -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: diff --git a/.github/workflows/self-push.yml b/.github/workflows/self-push.yml index 15035704d0a..423b0c8c693 100644 --- a/.github/workflows/self-push.yml +++ b/.github/workflows/self-push.yml @@ -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: diff --git a/.github/workflows/self-scheduled.yml b/.github/workflows/self-scheduled.yml index 09ea3af0d44..13d6a380698 100644 --- a/.github/workflows/self-scheduled.yml +++ b/.github/workflows/self-scheduled.yml @@ -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: