From 26750f1c2adc0d2c86721e1e9bebb59e6d5d300c Mon Sep 17 00:00:00 2001
From: Xuehai Pan <XuehaiPan@pku.edu.cn>
Date: Sun, 24 Jul 2022 19:23:29 +0800
Subject: [PATCH] chore(core/libnvml): update supported `nvidia-ml-py` version
 list

Signed-off-by: Xuehai Pan <XuehaiPan@pku.edu.cn>
---
 README.md         | 14 --------------
 nvitop/version.py |  2 ++
 pyproject.toml    |  2 +-
 requirements.txt  |  2 +-
 4 files changed, 4 insertions(+), 16 deletions(-)

diff --git a/README.md b/README.md
index b6e32aab..ef53af84 100644
--- a/README.md
+++ b/README.md
@@ -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
diff --git a/nvitop/version.py b/nvitop/version.py
index 1ffc5955..a8baea80 100644
--- a/nvitop/version.py
+++ b/nvitop/version.py
@@ -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>`_.
diff --git a/pyproject.toml b/pyproject.toml
index 1f7099ae..6cfbfe30 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -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",
diff --git a/requirements.txt b/requirements.txt
index 4ae7310f..8060624c 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -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