From 142089860238d4a577e56d208deb008cc736503c Mon Sep 17 00:00:00 2001 From: Alex Date: Mon, 16 Sep 2024 09:36:23 +0800 Subject: [PATCH] [CoreEngine] set the cuda visible id into the docker container when training. --- python/fedml/computing/scheduler/slave/base_slave_job_runner.py | 1 + 1 file changed, 1 insertion(+) diff --git a/python/fedml/computing/scheduler/slave/base_slave_job_runner.py b/python/fedml/computing/scheduler/slave/base_slave_job_runner.py index 5e530dbba7..9ea2c4beaf 100755 --- a/python/fedml/computing/scheduler/slave/base_slave_job_runner.py +++ b/python/fedml/computing/scheduler/slave/base_slave_job_runner.py @@ -251,6 +251,7 @@ def start_runner_process( agent_config=None, edge_id=edge_id ) client_runner.start_request_json = request_json + client_runner.cuda_visible_gpu_ids_str = cuda_visible_gpu_ids_str run_id_str = str(run_id) self.run_process_event = multiprocessing.Event() client_runner.run_process_event = self.run_process_event