diff --git a/isaacgymenvs/train.py b/isaacgymenvs/train.py
index 783a92d71..3ca00d5ef 100644
--- a/isaacgymenvs/train.py
+++ b/isaacgymenvs/train.py
@@ -140,6 +140,7 @@ def build_runner(algo_observer):
         return runner
 
     rlg_config_dict = omegaconf_to_dict(cfg.train)
+    rlg_config_dict['params']['config']['device']=cfg.rl_device
 
     # convert CLI arguments into dictionory
     # create runner and set the settings