add in environment variable to fix GPU memory allocation for tensorflow

NVIDIA-Merlin · Nov 21, 2022 · 16d4d64 · 16d4d64
1 parent 05060fc
commit 16d4d64
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ci/test_unit.sh b/ci/test_unit.sh
@@ -18,6 +18,6 @@
 set -e
 
 # Run tests
-pytest -rsx tests/unit tests/notebook
+TF_GPU_ALLOCATOR=cuda_malloc_async pytest -rsx tests/unit tests/notebook