diff --git a/pyproject.toml b/pyproject.toml index 869247f..c38bf36 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -44,7 +44,7 @@ classifiers = [ "Topic :: Utilities", ] dependencies = [ - "nvidia-ml-py >= 11.450.51, < 11.500.00", + "nvidia-ml-py >= 11.450.51, < 11.500.0a0", "psutil >= 5.6.6", "cachetools >= 1.0.1", "termcolor >= 1.0.0", diff --git a/requirements.txt b/requirements.txt index 4ae7310..0878e2c 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.500.0a0 psutil >= 5.6.6 cachetools >= 1.0.1 termcolor >= 1.0.0