How can I use different graphics cards for different RL tasks?

Hi there,
I have a server with 4 RTX3090 GPUs. When I try to run RL training tasks, the default GPU is CUDA:0 , I have typed “sim_device=cuda:1 rl_device=cuda:1 device_id=1” in the terminal commands, however in the terminal it shows : ‘RunTimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:1 and cuda:0!’
So, how can i run different tasks using different gpu devices?

Hi there, we have recently found a bug that is preventing the simulation from running on the correct GPU device. The fix will be included in the next Isaac Sim release. For now, you can try setting the CUDA_VISIBLE_DEVICES environment variable to specify the GPU you’d like to use.

1 Like

Thank you for your help! I was able to change a GPU device after changing the CUDA_VISIBLE_DEVICES environment.

This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.