Skip to content

Adding option to select GPU device for offscreen renderer#142

Merged
amandlek merged 3 commits intobug-fixesfrom
render_gpu_device
Dec 13, 2020
Merged

Adding option to select GPU device for offscreen renderer#142
amandlek merged 3 commits intobug-fixesfrom
render_gpu_device

Conversation

@amandlek
Copy link
Member

  • All environments now have an argument called render_gpu_device_id allowing for selection of GPU device to use for offscreen rendering. This is important for multi-GPU machines.

render_camera="frontview",
render_collision_mesh=False,
render_visual_mesh=True,
render_gpu_device_id=-1,
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I assume -1 corresponds to no GPU (or maybe default?) -- can we make this explicit in the docstring?

Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

yeah it's inferred according to this here - we can update the docstring

Copy link
Member

@cremebrule cremebrule left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

one small comment, otherwise LGTM

@amandlek amandlek merged commit 8be46b6 into bug-fixes Dec 13, 2020
@amandlek amandlek deleted the render_gpu_device branch December 13, 2020 22:58
X-jy-liu pushed a commit to X-jy-liu/robosuite that referenced this pull request Mar 5, 2025
…gpu_device

Adding option to select GPU device for offscreen renderer
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants