[PR #8] [MERGED] MIG device support #120

New issue

Closed

opened 2026-05-05 03:26:16 -06:00 by gitea-mirror · 0 comments

gitea-mirror commented

2026-05-05 03:26:16 -06:00

Owner

📋 Pull Request Information

Original PR: https://github.com/XuehaiPan/nvitop/pull/8
Author: @XuehaiPan
Created: 11/19/2021
Status: ✅ Merged
Merged: 6/26/2022
Merged by: @XuehaiPan

Base: dev ← Head: mig-support

📝 Commits (10+)

45542ca feat(core/device): add class MigDevice for MIG support
61f84a8 feat(core/device): update CUDA device enumeration
1f51739 feat(gui): MIG device support
1c5daec fix(core/device): fix variable reference
dda66a9 chore(core/process): set default process GPU utilization to N/A
463acc3 refactor(gui/device): rename variables
60802bf chore(core/device): move abstract methods to class Device
f9b1d4b refactor(core/device): cache physical device for MIG instances
2a1b856 fix(gui/device): fix MIG device snapshots
5c36522 style: apply suggestions from pylint

📊 Changes

17 files changed (+1402 additions, -293 deletions)

View changed files

📝 README.md (+183 -40)
➕ nvitop/callbacks/lightning.py (+6 -0)
➕ nvitop/callbacks/tensorboard.py (+12 -0)
📝 nvitop/callbacks/utils.py (+1 -1)
📝 nvitop/cli.py (+2 -1)
📝 nvitop/core/__init__.py (+4 -127)
➕ nvitop/core/collector.py (+683 -0)
📝 nvitop/core/device.py (+301 -42)
📝 nvitop/core/process.py (+19 -10)
📝 nvitop/gui/library/device.py (+55 -3)
📝 nvitop/gui/library/keybinding.py (+1 -1)
📝 nvitop/gui/screens/main/__init__.py (+2 -2)
📝 nvitop/gui/screens/main/device.py (+108 -41)
📝 nvitop/gui/screens/main/host.py (+2 -2)
📝 nvitop/gui/screens/main/process.py (+16 -16)
📝 nvitop/gui/screens/main/utils.py (+1 -1)
📝 nvitop/gui/screens/treeview.py (+6 -6)

📄 Description

Issue Type

Improvement/feature implementation

Runtime Environment

Operating system and version: Ubuntu 20.04 LTS
Terminal emulator and version: GNOME Terminal 3.36.2
Python version: 3.5+
NVML version (driver version): 430.64 / 460.84 / 470.82.00
nvitop version or commit: WIP
nvidia-ml-py version: 11.450.51 / 11.450.129 / 11.495.46
Locale: C / C.UTF-8 / en_US.UTF-8

Description

Add MIG device support to nvitop.

core/device: Add class MigDevice and update CUDA_VISIBLE_DEVICES handling for MIG devices.
gui: Update nvitop's UI for MIG enabled setup.

Motivation and Context

Add MIG device support to nvitop. Resolves #5.

Testing

Help wanted, see https://github.com/XuehaiPan/nvitop/pull/8#issuecomment-1155241507.

_{🔄 This issue represents a GitHub Pull Request. It cannot be merged through Gitea due to API limitations.}

## 📋 Pull Request Information **Original PR:** https://github.com/XuehaiPan/nvitop/pull/8 **Author:** [@XuehaiPan](https://github.com/XuehaiPan) **Created:** 11/19/2021 **Status:** ✅ Merged **Merged:** 6/26/2022 **Merged by:** [@XuehaiPan](https://github.com/XuehaiPan) **Base:** `dev` ← **Head:** `mig-support` --- ### 📝 Commits (10+) - [`45542ca`](https://github.com/XuehaiPan/nvitop/commit/45542ca426bf7255b8533cb0ff20ebd565539622) feat(core/device): add class MigDevice for MIG support - [`61f84a8`](https://github.com/XuehaiPan/nvitop/commit/61f84a893626128bdd40051d8a5d90f2d8966409) feat(core/device): update CUDA device enumeration - [`1f51739`](https://github.com/XuehaiPan/nvitop/commit/1f5173973d13bd26e99b631bba25cab342f1e7b9) feat(gui): MIG device support - [`1c5daec`](https://github.com/XuehaiPan/nvitop/commit/1c5daec2c0d0e94f7fae78e2c603e71ad02ee825) fix(core/device): fix variable reference - [`dda66a9`](https://github.com/XuehaiPan/nvitop/commit/dda66a9670ce6c3fa9aad328a334e67383ee48c4) chore(core/process): set default process GPU utilization to N/A - [`463acc3`](https://github.com/XuehaiPan/nvitop/commit/463acc3cc6bd92962214882a442aa7ab86ea2adc) refactor(gui/device): rename variables - [`60802bf`](https://github.com/XuehaiPan/nvitop/commit/60802bf734cf2f0a837d0f919317723956d715a5) chore(core/device): move abstract methods to class Device - [`f9b1d4b`](https://github.com/XuehaiPan/nvitop/commit/f9b1d4bdfbbbc363212d6a232c52df13cf491c5f) refactor(core/device): cache physical device for MIG instances - [`2a1b856`](https://github.com/XuehaiPan/nvitop/commit/2a1b856b0997812576a3c9c9e4c07c4cd23432f5) fix(gui/device): fix MIG device snapshots - [`5c36522`](https://github.com/XuehaiPan/nvitop/commit/5c365221681f9da7e44d99d42b53d637ec51952c) style: apply suggestions from pylint ### 📊 Changes **17 files changed** (+1402 additions, -293 deletions) <details> <summary>View changed files</summary> 📝 `README.md` (+183 -40) ➕ `nvitop/callbacks/lightning.py` (+6 -0) ➕ `nvitop/callbacks/tensorboard.py` (+12 -0) 📝 `nvitop/callbacks/utils.py` (+1 -1) 📝 `nvitop/cli.py` (+2 -1) 📝 `nvitop/core/__init__.py` (+4 -127) ➕ `nvitop/core/collector.py` (+683 -0) 📝 `nvitop/core/device.py` (+301 -42) 📝 `nvitop/core/process.py` (+19 -10) 📝 `nvitop/gui/library/device.py` (+55 -3) 📝 `nvitop/gui/library/keybinding.py` (+1 -1) 📝 `nvitop/gui/screens/main/__init__.py` (+2 -2) 📝 `nvitop/gui/screens/main/device.py` (+108 -41) 📝 `nvitop/gui/screens/main/host.py` (+2 -2) 📝 `nvitop/gui/screens/main/process.py` (+16 -16) 📝 `nvitop/gui/screens/main/utils.py` (+1 -1) 📝 `nvitop/gui/screens/treeview.py` (+6 -6) </details> ### 📄 Description #### Issue Type - Improvement/feature implementation #### Runtime Environment  - Operating system and version: Ubuntu 20.04 LTS - Terminal emulator and version: GNOME Terminal 3.36.2 - Python version: `3.5+` - NVML version (driver version): `430.64` / `460.84` / `470.82.00` - `nvitop` version or commit: WIP - `nvidia-ml-py` version: `11.450.51` / `11.450.129` / `11.495.46` - Locale: `C` / `C.UTF-8` / `en_US.UTF-8` #### Description Add MIG device support to `nvitop`. - `core/device`: Add class `MigDevice` and update `CUDA_VISIBLE_DEVICES` handling for MIG devices. - `gui`: Update `nvitop`'s UI for MIG enabled setup. #### Motivation and Context Add MIG device support to `nvitop`. Resolves #5. #### Testing Help wanted, see https://github.com/XuehaiPan/nvitop/pull/8#issuecomment-1155241507. --- <sub>🔄 This issue represents a GitHub Pull Request. It cannot be merged through Gitea due to API limitations.</sub>