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

Add `nvmlDeviceGetNvLink*` function wrappers #8

Closed rjzamora closed 5 years ago

rjzamora commented 5 years ago

As pointed out to me by @mt-jones, the NVML C API currently provides NvLink functions, like nvmlDeviceGetNvLinkUtilizationCounter.

PyNVML wrappers for these functions would be useful to multi-GPU users.

mrocklin commented 5 years ago

Thanks for the pointer @mt-jones !

mrocklin commented 5 years ago

Was this fixed by https://github.com/gpuopenanalytics/pynvml/pull/9 ?

rjzamora commented 5 years ago

@mrocklin Yes, this can be closed. However, the nvmlDeviceSetNvLinkUtilizationControl function (and maybe others) may need to be improved in the future. That is, I am curently using an nvidia-smi system call to produce a rough bokeh plot for NvLink usage. I would like to be able to use pure pynvml. The problem may be that the nvmlNvLinkUtilizationControl_t type is currently missused.