search

gpuopenanalytics / pynvml

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

Update NVML bindings (align with nvidia-ml-py) #32

Closed rjzamora closed 3 years ago

rjzamora commented 3 years ago

The NVML bindings have fallen "out of date" since this project was first published to conda/pypi. The nvidia-ml-py has since been updated to align with cuda 11. In the future, these efforts will likely converge under the "pynvml" name. For now, this PR simply uses the most recent changes in nvidia-ml-py to update the NVML bindings for this project.

rjzamora commented 3 years ago

@kenhester - I'd like to make a short-term update like this to unblock pynvml users on MIG/CUDA-11 (and to enable some NVDashboard fixes). Let me know if you have any concerns, or if you feel that this will create too many confilcts for the private branch you have planned to merge.