Cuda memory catch (#115)

Added a try catch so that in case of OOM errors on GPU it tries to clear gpu memory and continue when doing hyperparameter tuning.
OHDSI · Apr 29, 2024 · b95d31a · b95d31a
1 parent f343e72
commit b95d31a
Show file tree

Hide file tree

Showing 14 changed files with 385 additions and 201 deletions.
diff --git a/.github/workflows/R_CDM_check_hades.yaml b/.github/workflows/R_CDM_check_hades.yaml
@@ -78,7 +78,7 @@ jobs:
         shell: Rscript {0}
         run: |
           python_packages <- 
-            c("polars", "tqdm", "connectorx", "pyarrow", "scikit-learn")
+            c("polars", "tqdm", "connectorx", "pyarrow", "pynvml")
           
           library(reticulate)
           virtualenv_create("r-reticulate", Sys.which("python"))

diff --git a/DESCRIPTION b/DESCRIPTION
@@ -52,6 +52,7 @@ Config/reticulate:
       list(package = "polars"),
       list(package = "tqdm"),
       list(package = "connectorx"),
-      list(package = "pyarrow")
+      list(package = "pyarrow"),
+      list(package = "pynvml")
       )
   )