From 7dee9a189a6c3230e7605b49edff71d0126a376c Mon Sep 17 00:00:00 2001 From: jlnav Date: Mon, 12 Jul 2021 16:50:36 -0500 Subject: [PATCH] set CUDA_VISIBLE_DEVICES based on workerID --- libensemble/tests/scaling_tests/ddmd/keras_cvae_ml_genf.py | 1 + libensemble/tests/scaling_tests/ddmd/openmm_md_simf.py | 1 + 2 files changed, 2 insertions(+) diff --git a/libensemble/tests/scaling_tests/ddmd/keras_cvae_ml_genf.py b/libensemble/tests/scaling_tests/ddmd/keras_cvae_ml_genf.py index 2938478bf..51e1d4770 100644 --- a/libensemble/tests/scaling_tests/ddmd/keras_cvae_ml_genf.py +++ b/libensemble/tests/scaling_tests/ddmd/keras_cvae_ml_genf.py @@ -174,6 +174,7 @@ def run_keras_cvae_ml_genf(H, persis_info, gen_specs, libE_info): exctr = Executor.executor persis_info['stage_count'] = -1 os.environ["OMP_NUM_THREADS"] = '4' + os.environ["CUDA_VISIBLE_DEVICES"] = str(libE_info['workerID']) initial_complete = False tag = None diff --git a/libensemble/tests/scaling_tests/ddmd/openmm_md_simf.py b/libensemble/tests/scaling_tests/ddmd/openmm_md_simf.py index c97167fc6..3ca58f861 100644 --- a/libensemble/tests/scaling_tests/ddmd/openmm_md_simf.py +++ b/libensemble/tests/scaling_tests/ddmd/openmm_md_simf.py @@ -49,6 +49,7 @@ def run_openmm_sim_f(H, persis_info, sim_specs, libE_info): """ calc_status = 0 os.environ["OMP_NUM_THREADS"] = '4' + os.environ["CUDA_VISIBLE_DEVICES"] = str(libE_info['workerID']) # Specify the Executor object created in the calling script. exctr = Executor.executor