Can you drop dependency on nvidia-ml-py3?

[xwjiang2010]

Describe the Bug

A clear and concise description of what the bug is.

Both nvidia-ml-py3 and nvidia-ml are installing the same pynvml.py module, causing issues with the newest gpustat release. Error msg:

ImportError: pynvml is missing or an outdated version is installed.

Expected behavior

A clear and concise description of what you expected to happen.

Where is the issue?

• Comet Python SDK
• Comet UI
• Third Party Integrations (Huggingface, TensorboardX, Pytorch Lighting etc)

To Reproduce

Steps to reproduce the behavior:

1. Go to '...'
2. Click on '....'
3. Scroll down to '....'
4. See error

Stack Trace

If possible please include the full stack trace of your issue here

# Paste stack trace here

Comet Debug Log

If possible, please follow the instructions here to run Comet in debug mode and attach the resulting log file.

Screenshots or GIFs

If applicable, add screenshots/gifs to help explain your problem.

Link to Comet Project/Experiment

If applicable, please provide a link to your Comet Project or Experiment.

Additional context

Add any other context about the problem here.

[pcmoritz]

What is the stance of the comet maintainers on this issue? This is a blocker for adopting comet for anybody who wants to use a recent version of gpustat.

[pcmoritz]

I also linked the related gpustat issue

[DN6]

Hi all. Taking a look at this. Will get back to you ASAP

[mattip]

Any news here?

[mattip]

Hi. Is there an update to this issue?

[DN6]

Hi all. The dev team has identified the issue and found a viable workaround that will maintain current Comet functionality while fixing this issue. The fix will be released within the next two weeks.

[jacques-comet]

Hi @mattip @xwjiang2010
The latest version of the Comet SDK (version 3.31.9) has dropped the nvidia-ml-py3 dependency, let me know if you are still running into any issues on the latest version of the SDK

[jacques-comet]

From version 3.31.9 onwards nvidia-ml-py3 is no longer a dependency of the SDK. Closing this issue as resolved

[mattip]

Thanks!