[ci] Ensure that TF does not preempt all GPU memory for itself

see https://www.tensorflow.org/guide/gpu#limiting_gpu_memory_growth

Co-Authored-By: Funtowicz Morgan <mfuntowicz@users.noreply.github.com>
Co-Authored-By: Lysandre Debut <lysandre.debut@reseau.eseo.fr>
This commit is contained in:
Julien Chaumond
2020-03-02 11:56:45 -05:00
parent c0135194eb
commit 13afb71208
2 changed files with 2 additions and 0 deletions

View File

@@ -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: |

View File

@@ -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