Convert pytorch_lighting imports into lightning.pytorch imports

XuehaiPan / nvitop

An interactive NVIDIA-GPU process viewer and beyond, the one-stop solution for GPU process management.

https://nvitop.readthedocs.io

Apache License 2.0

4.56k stars 144 forks source link

Convert pytorch_lighting imports into lightning.pytorch imports #113

Closed marios1861 closed 8 months ago

marios1861 commented 8 months ago

Bug fix

Description

Fixes #112

Testing

Isolated change, working as intended.

XuehaiPan commented 8 months ago

Hi @marios1861 thanks for the pull request. Following the discussion in issue #112, I would rather keep the support for pytorch-lightning. Because pytorch-lightning and lightning are different packages on PyPI, and they are not compatible with each other (see your issue). I populate the implementation for lightning based on lighting.pytorch.callbacks.Callback in PR #114.