search

openai / mujoco-py

MuJoCo is a physics engine for detailed, efficient rigid body simulations with contacts. mujoco-py allows using MuJoCo from Python 3.
Other
2.79k stars 810 forks source link

Change the device for rendering #778

Open wadiuvatzy opened 8 months ago

wadiuvatzy commented 8 months ago

I am using mujoco-py to do some RL tasks. When I am runing the program, it will output "Found 8 GPUs for rendering. Using device 0" image When I want to get more processes, cuda: 0 is out of memory. So I want to change the default device for rendering but I donot know how.

I have tried to add "CUDA_VISIBLE_DEVICES=1" before the python command. But there is another error coming out.

9d042071809c144b6170f4296269037

I have no idea how to fix this.

So could you tell me how to change the device for rendering?