[Feature Request] Add support to AMD's ROCm GPU

Required prerequisites

• I have searched the Issue Tracker that this hasn't already been reported. (comment there if it has.)
• I have tried the latest version of nvitop in a new isolated virtual environment.

Motivation

I have been using nvitop for monitoring NVIDIA devices and processes, and I find it to be a great tool with a beautiful UI. Thank you for this good project!

However, I noticed that it doesn't support AMD's ROCm GPU platform. As an AMD user (we have an AMD GPU cluster), I can only use "rocm-smi" to monitor my GPU, and I would love to have a similar tool like nvitop for ROCm.

I believe that adding support for AMD's ROCm GPU would make nvitop a more versatile and inclusive monitoring tool. It would allow users who work with AMD GPUs to benefit from the same features and options that nvitop provides to NVIDIA users.

Solution

Using rocm-smi

Alternatives

No response

Additional context

No response