Merge pull request #172 from janrope/enable_numgpus_ray_init

[bug fix] added num_gpus variable to ray init call
Replicable-MARL · Aug 22, 2023 · 63bd131 · 63bd131
2 parents 921b72b + 5917868
commit 63bd131
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/marllib/marl/algos/run_cc.py b/marllib/marl/algos/run_cc.py
@@ -59,7 +59,7 @@ def restore_config_update(exp_info, run_config, stop_config):
 
 
 def run_cc(exp_info, env, model, stop=None):
-    ray.init(local_mode=exp_info["local_mode"])
+    ray.init(local_mode=exp_info["local_mode"], num_gpus=exp_info["num_gpus"])
 
     ########################
     ### environment info ###

diff --git a/marllib/marl/algos/run_il.py b/marllib/marl/algos/run_il.py
@@ -34,7 +34,7 @@
 
 
 def run_il(exp_info, env, model, stop=None):
-    ray.init(local_mode=exp_info["local_mode"])
+    ray.init(local_mode=exp_info["local_mode"], num_gpus=exp_info["num_gpus"])
 
     ########################
     ### environment info ###

diff --git a/marllib/marl/algos/run_vd.py b/marllib/marl/algos/run_vd.py
@@ -36,7 +36,7 @@
 
 
 def run_vd(exp_info, env, model, stop=None):
-    ray.init(local_mode=exp_info["local_mode"])
+    ray.init(local_mode=exp_info["local_mode"], num_gpus=exp_info["num_gpus"])
 
     ########################
     ### environment info ###