chore: add notes for syncing nvidia-ml-py versions

Signed-off-by: Xuehai Pan <XuehaiPan@pku.edu.cn>
This commit is contained in:
Xuehai Pan 2022-10-17 18:42:13 +08:00
parent 8e9a3f8d2d
commit dbf018beca
3 changed files with 3 additions and 0 deletions

View file

@ -44,6 +44,7 @@ if not __release__:
# 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 = [
# 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
'11.460.79',

View file

@ -44,6 +44,7 @@ classifiers = [
"Topic :: Utilities",
]
dependencies = [
# Sync with nvitop/version.py and requirements.txt
"nvidia-ml-py >= 11.450.51, < 11.516.0a0",
"psutil >= 5.6.6",
"cachetools >= 1.0.1",

View file

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