Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

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

Open
Charlie0257 opened this issue Jan 4, 2024 · 2 comments
Open

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

Charlie0257 opened this issue Jan 4, 2024 · 2 comments

Comments

@Charlie0257
Copy link

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?
@Denys88
Copy link
Owner

Denys88 commented Jan 10, 2024

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

@Charlie0257
Copy link
Author

@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?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@Denys88 @Charlie0257