rl_games icon indicating copy to clipboard operation
rl_games copied to clipboard
verified publisher icon Denys88

How to specify the GPU that loads the model weights when applying to IsaacGym

Open Charlie0257 opened this issue 2 years ago • 2 comments

Since I trained the corresponding model weights on the server, how do I specify the corresponding loaded GPU ID on my own single GPU machine?

Charlie0257 avatar Jan 04 '24 07:01 Charlie0257

you can setup device in yaml config. But by default it is just 'cuda' so it should choose your only device.

Denys88 avatar Jan 10 '24 14:01 Denys88

@Denys88 Thanks for your reply!

Setting up device in yaml config does not work!

The reason is return safe_filesystem_op(torch.load, filename) in torch_ext.py don't specify device!

Do you have any suggestions?

Charlie0257 avatar Feb 01 '24 08:02 Charlie0257