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

Restructuring and adding testing #6

Closed rjzamora closed 5 years ago

rjzamora commented 5 years ago

Restructuring, in response to issue#5 (https://github.com/gpuopenanalytics/pynvml/issues/5). I also added some simple testing and changed the names of several classes/methods in pynvml/nvml.py to be more "pythonic". Most of the nvml wrapper functions still have the same name as the C library, so more work is needed if we want a conventional python interface.

rjzamora commented 5 years ago

@mrocklin Okay - I think I addressed the testing. Also, the pynvml and nvida_smi modules can now be loaded and used exactly as before. We will probably want to improve the way the user interacts with these modules in the future, but I think we should let this PR focus on restructuring and testing.

mrocklin commented 5 years ago

Thanks @rjzamora . This is in!