add in environment variable to fix GPU memory allocation for tensorflow

jperez999 · jperez999 · commit 16d4d64d4770 · 2022-11-21T17:47:28.000-05:00
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
 
 
