gpuopenanalytics / pynvml

Provide Python access to the NVML library for GPU diagnostics
BSD 3-Clause "New" or "Revised" License
212 stars 30 forks source link

Add loud warnings to encourage nvidia-ml-py migration #55

Closed rjzamora closed 2 months ago

rjzamora commented 2 months ago

There are no plans to maintain the (unofficial) pynvml package. This PR is an attempt to encourage faster user migration to the official nvidia-ml-py package:

@kenhester - I would prefer for the pynvml.smi warning to point the user somewhere else for smi support. Let me know if you (or anyone else) are motivated to create and maintain such space. Unfortunately, there is no change for that space to be within nvidia-ml-py.

rjzamora commented 2 months ago

Update...@pentschev pointed out in an offline discussion that it will be difficult for all RAPIDS-related projects to migrate to nvidia-ml-py in a coordinated way (which would be necessary since both packages are imported as pynvml). He suggested that we convert this repo to a "meta package" that simply adds nvidia-ml-py as a dependency. If we are taking the time to refactor the project in this way, it could also make sense to move the smi bindings into a distinct pynvml_utils module before the pynvml module is removed.

rjzamora commented 2 months ago

Merging the README change, and moving the smi refactor to a new PR.

pentschev commented 2 months ago

Thanks Rick!