chore!: remove per-version install extras for nvidia-ml-py and prefer nvitop[cudaXX] instead (#179)

.pre-commit-config.yaml

@@ -26,7 +26,7 @@ repos:
         fail_fast: true
       - id: debug-statements
   - repo: https://github.com/astral-sh/ruff-pre-commit
-    rev: v0.12.8
+    rev: v0.12.9
     hooks:
       - id: ruff-check
         args: [--fix, --exit-non-zero-on-fix]

CHANGELOG.md

@@ -25,7 +25,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ### Removed
 
--
+- Remove per-version install extras for `nvidia-ml-py` and prefer `nvitop[cudaXX]` instead by [@XuehaiPan](https://github.com/XuehaiPan) in [#179](https://github.com/XuehaiPan/nvitop/pull/179).
 
 ------
 

docs/source/spelling_wordlist.txt

@@ -17,6 +17,7 @@ csv
 ctrl
 ctx
 cuda
+cudaxx
 cudadevice
 cudaerror
 cwd

nvitop/version.py

@@ -95,17 +95,17 @@ PYNVML_VERSION_CANDIDATES = (
 """The list of supported ``nvidia-ml-py`` versions.
 See also: `nvidia-ml-py's Release History <https://pypi.org/project/nvidia-ml-py/#history>`_.
 
-To install ``nvitop`` with a specific version of ``nvidia-ml-py``, use ``nvitop[pynvml-xx.yyy.zzz]``, for example:
+To install ``nvitop`` with a specific version of ``nvidia-ml-py``, use:
 
 .. code:: bash
 
-    pip3 install 'nvitop[pynvml-11.450.51]'
+    pip3 install nvidia-ml-py==xx.yyy.zz nvitop
 
 or
 
 .. code:: bash
 
-    pip3 install nvitop nvidia-ml-py==11.450.51
+    pip3 install 'nvitop[cudaXX]'
 
 Note:
     The package ``nvidia-ml-py`` is not backward compatible over releases. This may cause problems

setup.py

@@ -6,7 +6,7 @@
 #
 # or
 #
-#   pip install 'nvitop[pynvml-xx.yyy.zz]'
+#   pip install 'nvitop[cudaXX]'
 #
 
 """Setup script for ``nvitop``."""
@@ -101,13 +101,6 @@ if __name__ == '__main__':
                 extra_requirements[f'cuda{pynvml_major}'] = [
                     f'nvidia-ml-py >= {pynvml_range[0]}, <= {pynvml_range[-1]}',
                 ]
-        extra_requirements.update(
-            {
-                # The identifier could not start with numbers, add a prefix `pynvml-`
-                f'pynvml-{pynvml}': [f'nvidia-ml-py == {pynvml}']
-                for pynvml in version.PYNVML_VERSION_CANDIDATES
-            },
-        )
 
         setup(
             name='nvitop',