llm-foundry icon indicating copy to clipboard operation
llm-foundry copied to clipboard

Use the nvidia-supplied nvidia-ml-py instead of pynvml

Open mattip opened this issue 2 years ago • 0 comments

This project depends on pynvml, where the nvidia-blessed bindings to nvml are in the nvidia-ml-py package instead. This is causing problems for the gpustat project, which needs some internals from nvidia-ml-py to properly support older nvml version. Would it be possible to switch?

mattip avatar May 17 '23 09:05 mattip