github-starred/nvitop
2
0
Fork 0
mirror of https://github.com/XuehaiPan/nvitop.git synced 2026-05-22 06:05:47 -06:00
Code Issues 29 Projects Packages Wiki Activity Actions
nvitop/examples/select-devices-api
History
Exact
Xuehai Pan 766906266b chore: update license header
2026-05-16 15:37:30 +08:00
..
README.md feat(examples): add examples/ folder and relocate the deprecated nvitop.callbacks (#216) 2026-05-16 15:34:29 +08:00
select_devices_api.py chore: update license header 2026-05-16 15:37:30 +08:00

README.md

Programmatic CUDA Device Selection

Shows nvitop.select_devices — the Python API behind the nvisel CLI. Useful for scripts that need to pick GPUs based on free memory, utilization, or other criteria before launching a training job.

APIs Used

  • nvitop.select_devices — supports format='index' | 'uuid' | 'device', min_count, min_free_memory, max_gpu_utilization, and more.

Run

python3 examples/select-devices-api/select_devices_api.py

Requires only nvitop itself; no other dependencies.

See ../README.md for the full example index.