Skip to content

Commit

Permalink
chore(core/libnvml): update supported nvidia-ml-py version list
Browse files Browse the repository at this point in the history
Signed-off-by: Xuehai Pan <XuehaiPan@pku.edu.cn>
  • Loading branch information
XuehaiPan committed Jul 25, 2022
1 parent 70dc9a7 commit 26750f1
Show file tree
Hide file tree
Showing 4 changed files with 4 additions and 16 deletions.
14 changes: 0 additions & 14 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -162,20 +162,6 @@ pip3 install .
</br>
</p>

**IMPORTANT:** `pip` will install `nvidia-ml-py>=11.450.51,<=11.495.46` as a dependency for `nvitop`. Please verify whether the `nvidia-ml-py` package is compatible with your NVIDIA driver version. You can check the release history of `nvidia-ml-py` at [nvidia-ml-py's Release History](https://pypi.org/project/nvidia-ml-py/11.495.46/#history), and install the compatible version manually by:

```bash
pip3 install --no-dependencies 'nvidia-ml-py==xx.yyy.zzz'
```

Since `nvidia-ml-py>=11.450.129`, the definition of `nvmlProcessInfo_t` has introduced two new fields `gpuInstanceId` and `computeInstanceId` (`GI ID` and `CI ID` in newer `nvidia-smi`) which are incompatible with some old NVIDIA drivers. `nvitop` may not display the processes correctly due to this incompatibility.

You can specified the version of `nvidia-ml-py` while installing `nvitop` as:

```bash
pip3 install 'nvitop[pynvml-11.450.51]' # or 'nvitop[cuda10]'
```

------

## Usage
Expand Down
2 changes: 2 additions & 0 deletions nvitop/version.py
Original file line number Diff line number Diff line change
Expand Up @@ -39,6 +39,8 @@
'11.460.79',
'11.470.66',
'11.495.46',
'11.510.69',
'11.515.48',
]
"""The list of supported ``nvidia-ml-py`` versions.
See also: `nvidia-ml-py's Release History <https://pypi.org/project/nvidia-ml-py/#history>`_.
Expand Down
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -37,7 +37,7 @@ classifiers = [
"Topic :: Utilities",
]
dependencies = [
"nvidia-ml-py >= 11.450.51, < 11.500.00",
"nvidia-ml-py >= 11.450.51, < 11.516.0a0",
"psutil >= 5.6.6",
"cachetools >= 1.0.1",
"termcolor >= 1.0.0",
Expand Down
2 changes: 1 addition & 1 deletion requirements.txt
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
nvidia-ml-py >= 11.450.51, < 11.500.00
nvidia-ml-py >= 11.450.51, < 11.516.0a0
psutil >= 5.6.6
cachetools >= 1.0.1
termcolor >= 1.0.0
Expand Down

0 comments on commit 26750f1

Please sign in to comment.