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.81k stars 810 forks source link

Confusion about the number of GPUs #716

Open zichunxx opened 2 years ago

zichunxx commented 2 years ago

When I tried to render the Mujoco model with a specified GPU, I got some confusion, which may be very simple.

When the following code is executed, I can find Found 3 GPUs for rendering. Using device 0 in the Pycharm console window.

image = sim.render(width=300, height=300, camera_name = 'cam0', depth=False, device_id=0)

However, when I use pytorch to detect the number and name of GPU, the related codes and displays are shown below

print('Number of gpu: ', torch.cuda.device_count())
print('its name： ', torch.cuda.get_device_name(0))

Number of gpu: 1
its name：  NVIDIA GeForce RTX 3070 Ti

(1) Why the numbers of GPUs detected by the two methods are different?

(2) For Mujoco, how can I check the names of the detected GPUs?

(3) After I installed the Nvidia driver, the default installation path was /usr/src instead of /usr/lib. Then, I use ln -s to create a new folder /usr/lib/nvidia-510. However, there is no libGL.so mentioned in some other issues.

Hope to get some replies!

kano00 commented 2 years ago

Hi. I have the same problem. Have you found the answer?

zichunxx commented 2 years ago

Hi. I have the same problem. Have you found the answer?

Sry, no idea right now...

BetterZH commented 1 year ago

I have the same problem. Have you found the answer?