chore(core/libnvml): update supported nvidia-ml-py version list

Signed-off-by: Xuehai Pan <XuehaiPan@pku.edu.cn>
This commit is contained in:
Xuehai Pan 2022-07-24 19:23:29 +08:00
parent b88ecfe2a5
commit 9c9b02809a
4 changed files with 4 additions and 16 deletions

View file

@ -166,20 +166,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

View file

@ -49,6 +49,8 @@ PYNVML_VERSION_CANDIDATES = [
'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>`_.

View file

@ -44,7 +44,7 @@ classifiers = [
"Topic :: Utilities",
]
dependencies = [
"nvidia-ml-py >= 11.450.51, < 11.500.0a0",
"nvidia-ml-py >= 11.450.51, < 11.516.0a0",
"psutil >= 5.6.6",
"cachetools >= 1.0.1",
"termcolor >= 1.0.0",

View file

@ -1,4 +1,4 @@
nvidia-ml-py >= 11.450.51, < 11.500.0a0
nvidia-ml-py >= 11.450.51, < 11.516.0a0
psutil >= 5.6.6
cachetools >= 1.0.1
termcolor >= 1.0.0