Skip to content

Feature request: pass a list of GPU ids #55

@erwincoumans

Description

@erwincoumans

Currently, for multi-GPU it looks like MimicKit hard-codes [cuda:0 .. num_gpu). It would be nice to explicitly pass specific GPU ids (say cuda:1, cuda:3).
Similar to mjlab:

uv run train Mjlab-Velocity-Flat-Unitree-G1 --gpu-ids 1 3 

Metadata

Metadata

Assignees

Labels

No labels
No labels

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions