Merge pull request #2226 from FedML-AI/alexleung/dev_v0700_for_merge

[CoreEngine] set the cuda visible id into the docker container when t…
FedML-AI · Sep 16, 2024 · d0826d9 · d0826d9
2 parents fa11d0b + 1420898
commit d0826d9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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