diff --git a/.github/workflows/self-push.yml b/.github/workflows/self-push.yml index cfbe999699..e21520d0e8 100644 --- a/.github/workflows/self-push.yml +++ b/.github/workflows/self-push.yml @@ -40,6 +40,7 @@ jobs: - name: Run all non-slow tests on GPU env: + TF_FORCE_GPU_ALLOW_GROWTH: yes OMP_NUM_THREADS: 1 USE_CUDA: yes run: | diff --git a/.github/workflows/self-scheduled.yml b/.github/workflows/self-scheduled.yml index 7c33d5dfcb..d963ab5920 100644 --- a/.github/workflows/self-scheduled.yml +++ b/.github/workflows/self-scheduled.yml @@ -41,6 +41,7 @@ jobs: - name: Run all tests on GPU env: + TF_FORCE_GPU_ALLOW_GROWTH: yes OMP_NUM_THREADS: 1 RUN_SLOW: yes USE_CUDA: yes