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 5e530dbba..9ea2c4bea 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