deps: bump latest supported nvidia-ml-py version

Signed-off-by: Xuehai Pan <XuehaiPan@pku.edu.cn>
This commit is contained in:
Xuehai Pan 2023-01-11 15:25:54 +08:00
parent cd735eb2df
commit e5528f4ca3
3 changed files with 5 additions and 4 deletions

View file

@ -56,7 +56,7 @@ if not __release__:
# The package `nvidia-ml-py` is not backward compatible over releases. This may
# cause problems with Old versions of NVIDIA drivers.
# The ideal solution is to let the user install the best-fit version of `nvidia-ml-py`.
PYNVML_VERSION_CANDIDATES = [
PYNVML_VERSION_CANDIDATES = (
# Sync with pyproject.toml and requirements.txt
'11.450.51', # the last version supports the R430 driver (CUDA 10.x)
'11.450.129', # requires at last the R450 driver
@ -66,7 +66,8 @@ PYNVML_VERSION_CANDIDATES = [
'11.510.69', # the first version supports the `nvmlMemory_v2` API
'11.515.48',
'11.515.75',
]
'11.525.84',
)
"""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

@ -46,7 +46,7 @@ classifiers = [
]
dependencies = [
# Sync with nvitop/version.py and requirements.txt
"nvidia-ml-py >= 11.450.51, < 11.516.0a0",
"nvidia-ml-py >= 11.450.51, < 11.526.0a0",
"psutil >= 5.6.6",
"cachetools >= 1.0.1",
"termcolor >= 1.0.0",

View file

@ -1,5 +1,5 @@
# Sync with pyproject.toml and nvitop/version.py
nvidia-ml-py >= 11.450.51, < 11.516.0a0
nvidia-ml-py >= 11.450.51, < 11.526.0a0
psutil >= 5.6.6
cachetools >= 1.0.1
termcolor >= 1.0.0