[PR #24] [MERGED] Add extra options to specify `nvidia-ml-py` version #124

New issue

Closed

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

gitea-mirror commented

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

Owner

📋 Pull Request Information

Original PR: https://github.com/XuehaiPan/nvitop/pull/24
Author: @XuehaiPan
Created: 7/4/2022
Status: ✅ Merged
Merged: 7/5/2022
Merged by: @XuehaiPan

Base: dev ← Head: pynvml

📝 Commits (2)

9406192 feat(setup): add extra options to specify nvidia-ml-py version
b214e0a chore(cli): add message for missing functions on CUDA 10.x driver

📊 Changes

8 files changed (+96 additions, -12 deletions)

View changed files

📝 README.md (+8 -2)
📝 docs/source/apis/index.rst (+6 -0)
📝 nvitop/cli.py (+20 -4)
📝 nvitop/core/libnvml.py (+3 -3)
📝 nvitop/version.py (+39 -0)
📝 requirements.txt (+1 -1)
📝 setup.cfg (+3 -2)
📝 setup.py (+16 -0)

📄 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.9.13
NVML version (driver version): 470.129.06
nvitop version or commit: dev@2906d4f0
python-ml-py version: 11.450.51
Locale: en_US.UTF-8

Description

Unpin the package nvidia-ml-py version. By default, pip will install dependency nvidia-ml-py >= 11.450.51, < 11.510.69, or skip if already installed. To specify the nvidia-ml-py version with nvitop. Use:

pip3 install 'nvitop[pynvml-11.450.51]'

pip3 install nvitop nvidia-ml-py==11.495.46

An extra alias is added for CUDA 10.x drivers as:

pip3 install 'nvitop[cuda10]'  # installs nvidia-ml-py==11.450.51

The users will get an error message if nvitop is not compatible with R430 driver on Ubuntu 16.04 LTS:

The supported nvidia-ml-py versions are listed here:

8c3e727719/nvitop/version.py (L27-L63)

Ref: nvidia-ml-py's Release History.

Resolves #23.

Motivation and Context

See issue #23.

Testing

N/A

_{🔄 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/24 **Author:** [@XuehaiPan](https://github.com/XuehaiPan) **Created:** 7/4/2022 **Status:** ✅ Merged **Merged:** 7/5/2022 **Merged by:** [@XuehaiPan](https://github.com/XuehaiPan) **Base:** `dev` ← **Head:** `pynvml` --- ### 📝 Commits (2) - [`9406192`](https://github.com/XuehaiPan/nvitop/commit/94061923028b621a41e19d279e7a2d323266ccd7) feat(setup): add extra options to specify `nvidia-ml-py` version - [`b214e0a`](https://github.com/XuehaiPan/nvitop/commit/b214e0a713697c0b2c9eb7a23bf1e2694d0da17b) chore(cli): add message for missing functions on CUDA 10.x driver ### 📊 Changes **8 files changed** (+96 additions, -12 deletions) <details> <summary>View changed files</summary> 📝 `README.md` (+8 -2) 📝 `docs/source/apis/index.rst` (+6 -0) 📝 `nvitop/cli.py` (+20 -4) 📝 `nvitop/core/libnvml.py` (+3 -3) 📝 `nvitop/version.py` (+39 -0) 📝 `requirements.txt` (+1 -1) 📝 `setup.cfg` (+3 -2) 📝 `setup.py` (+16 -0) </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.9.13` - NVML version (driver version): `470.129.06` - `nvitop` version or commit: `dev@2906d4f0` - `python-ml-py` version: `11.450.51` - Locale: `en_US.UTF-8` #### Description  Unpin the package `nvidia-ml-py` version. By default, `pip` will install dependency `nvidia-ml-py >= 11.450.51, < 11.510.69`, or skip if already installed. To specify the `nvidia-ml-py` version with `nvitop`. Use: ```bash pip3 install 'nvitop[pynvml-11.450.51]' ``` or ```bash pip3 install nvitop nvidia-ml-py==11.495.46 ``` An extra alias is added for CUDA 10.x drivers as: ```bash pip3 install 'nvitop[cuda10]' # installs nvidia-ml-py==11.450.51 ``` The users will get an error message if `nvitop` is not compatible with R430 driver on Ubuntu 16.04 LTS: ![warnings](https://user-images.githubusercontent.com/16078332/177192031-f353f68d-b923-456a-815b-16936c0ad0e9.png) The supported `nvidia-ml-py` versions are listed here: https://github.com/XuehaiPan/nvitop/blob/8c3e727719a31d46b5050ed475b647cfca2565cc/nvitop/version.py#L27-L63 Ref: [nvidia-ml-py's Release History](https://pypi.org/project/nvidia-ml-py/11.495.46/#history). Resolves #23. #### Motivation and Context    See issue #23. #### Testing N/A --- <sub>🔄 This issue represents a GitHub Pull Request. It cannot be merged through Gitea due to API limitations.</sub>