mujoco-py icon indicating copy to clipboard operation
mujoco-py copied to clipboard

Confusion about the number of GPUs

Open zichunxx opened this issue 3 years ago • 3 comments

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!

zichunxx avatar May 14 '22 02:05 zichunxx

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

kano00 avatar Jul 18 '22 09:07 kano00

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

Sry, no idea right now...

zichunxx avatar Jul 19 '22 14:07 zichunxx

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

BetterZH avatar Nov 07 '22 07:11 BetterZH