llm-foundry
llm-foundry copied to clipboard
Use the nvidia-supplied nvidia-ml-py instead of pynvml
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?