[GH-ISSUE #74] [Feature Request] Add support to AMD's ROCm GPU #43

Open
opened 2026-05-05 03:23:12 -06:00 by gitea-mirror · 14 comments
Owner

Originally created by @Junyi-99 on GitHub (Jun 11, 2023).
Original GitHub issue: https://github.com/XuehaiPan/nvitop/issues/74

Originally assigned to: @XuehaiPan on GitHub.

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

Originally created by @Junyi-99 on GitHub (Jun 11, 2023). Original GitHub issue: https://github.com/XuehaiPan/nvitop/issues/74 Originally assigned to: @XuehaiPan on GitHub. ### Required prerequisites - [X] I have searched the [Issue Tracker](https://github.com/XuehaiPan/nvitop/issues) that this hasn't already been reported. (comment there if it has.) - [X] 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_
gitea-mirror added the
enhancement
api
upstream
labels 2026-05-05 03:23:12 -06:00
Author
Owner

@XuehaiPan commented on GitHub (Jun 21, 2023):

Hi @Junyi-99, thanks for raising this, and I apologize for the late response. I'm afraid that AMD graphics cards may not be supported in nvitop in the foreseeable future. This is due to a lack of necessary Python dependencies on PyPI (there is only an example in RadeonOpenCompute/rocm_smi_lib) and I personally don't have access to a machine with an AMD graphics card for testing. Sorry about that.

nvitop is completely open-source, and you are welcome to fork and develop your own monitor tool.

<!-- gh-comment-id:1600756935 --> @XuehaiPan commented on GitHub (Jun 21, 2023): Hi @Junyi-99, thanks for raising this, and I apologize for the late response. I'm afraid that AMD graphics cards may not be supported in `nvitop` in the foreseeable future. This is due to a lack of necessary Python dependencies on PyPI (there is only an example in [RadeonOpenCompute/rocm_smi_lib](https://github.com/RadeonOpenCompute/rocm_smi_lib/tree/master/python_smi_tools)) and I personally don't have access to a machine with an AMD graphics card for testing. Sorry about that. `nvitop` is completely open-source, and you are welcome to fork and develop your own monitor tool.
Author
Owner

@Junyi-99 commented on GitHub (Jun 21, 2023):

Thanks for your reply and for digging into ROCm monitoring solutions. I will continue to pay attention to this project, I think this issue can be closed.

Thank you again.

<!-- gh-comment-id:1601096304 --> @Junyi-99 commented on GitHub (Jun 21, 2023): Thanks for your reply and for digging into ROCm monitoring solutions. I will continue to pay attention to this project, I think this issue can be closed. Thank you again.
Author
Owner

@yan-rui commented on GitHub (Jun 26, 2023):

Hi @Junyi-99, thanks for raising this, and I apologize for the late response. I'm afraid that AMD graphics cards may not be supported in nvitop in the foreseeable future. This is due to a lack of necessary Python dependencies on PyPI (there is only an example in RadeonOpenCompute/rocm_smi_lib) and I personally don't have access to a machine with an AMD graphics card for testing. Sorry about that.

nvitop is completely open-source, and you are welcome to fork and develop your own monitor tool.

Thank you so much for this work. It has become an important part of my work.
But recently I started working on other GPUs and it's very inconvenient without nvitop.
Can you help point out what parts needs to be modified to replace nvidia-smi related commands?

<!-- gh-comment-id:1606710804 --> @yan-rui commented on GitHub (Jun 26, 2023): > Hi @Junyi-99, thanks for raising this, and I apologize for the late response. I'm afraid that AMD graphics cards may not be supported in `nvitop` in the foreseeable future. This is due to a lack of necessary Python dependencies on PyPI (there is only an example in [RadeonOpenCompute/rocm_smi_lib](https://github.com/RadeonOpenCompute/rocm_smi_lib/tree/master/python_smi_tools)) and I personally don't have access to a machine with an AMD graphics card for testing. Sorry about that. > > `nvitop` is completely open-source, and you are welcome to fork and develop your own monitor tool. Thank you so much for this work. It has become an important part of my work. But recently I started working on other GPUs and it's very inconvenient without nvitop. Can you help point out what parts needs to be modified to replace nvidia-smi related commands?
Author
Owner

@Junyi-99 commented on GitHub (Jun 26, 2023):

Hi @Junyi-99, thanks for raising this, and I apologize for the late response. I'm afraid that AMD graphics cards may not be supported in nvitop in the foreseeable future. This is due to a lack of necessary Python dependencies on PyPI (there is only an example in RadeonOpenCompute/rocm_smi_lib) and I personally don't have access to a machine with an AMD graphics card for testing. Sorry about that.
nvitop is completely open-source, and you are welcome to fork and develop your own monitor tool.

Thank you so much for this work. It has become an important part of my work. But recently I started working on other GPUs and it's very inconvenient without nvitop. Can you help point out what parts needs to be modified to replace nvidia-smi related commands?

What do you mean "to replace nvidia-smi related commands"?

<!-- gh-comment-id:1606735948 --> @Junyi-99 commented on GitHub (Jun 26, 2023): > > Hi @Junyi-99, thanks for raising this, and I apologize for the late response. I'm afraid that AMD graphics cards may not be supported in `nvitop` in the foreseeable future. This is due to a lack of necessary Python dependencies on PyPI (there is only an example in [RadeonOpenCompute/rocm_smi_lib](https://github.com/RadeonOpenCompute/rocm_smi_lib/tree/master/python_smi_tools)) and I personally don't have access to a machine with an AMD graphics card for testing. Sorry about that. > > `nvitop` is completely open-source, and you are welcome to fork and develop your own monitor tool. > > Thank you so much for this work. It has become an important part of my work. But recently I started working on other GPUs and it's very inconvenient without nvitop. Can you help point out what parts needs to be modified to replace nvidia-smi related commands? What do you mean "to replace nvidia-smi related commands"?
Author
Owner

@yan-rui commented on GitHub (Jun 26, 2023):

我现在用的GPU使用的监控工具与nvidia很像。
所以我想,是否需要将nvidia监控工具相关的api替换为现在在用的GPU的监控工具api,就可以用上nvitop了?

<!-- gh-comment-id:1606760842 --> @yan-rui commented on GitHub (Jun 26, 2023): 我现在用的GPU使用的监控工具与nvidia很像。 所以我想,是否需要将nvidia监控工具相关的api替换为现在在用的GPU的监控工具api,就可以用上nvitop了?
Author
Owner

@Junyi-99 commented on GitHub (Jun 26, 2023):

我现在用的GPU使用的监控工具与nvidia很像。 所以我想,是否需要将nvidia监控工具相关的api替换为现在在用的GPU的监控工具api,就可以用上nvitop了?

我认为是可以的

<!-- gh-comment-id:1606765958 --> @Junyi-99 commented on GitHub (Jun 26, 2023): > 我现在用的GPU使用的监控工具与nvidia很像。 所以我想,是否需要将nvidia监控工具相关的api替换为现在在用的GPU的监控工具api,就可以用上nvitop了? 我认为是可以的
Author
Owner

@yan-rui commented on GitHub (Jun 26, 2023):

我现在用的GPU使用的监控工具与nvidia很像。 所以我想,是否需要将nvidia监控工具相关的api替换为现在在用的GPU的监控工具api,就可以用上nvitop了?

我认为是可以的

我修改哪些地方可以达到这个目的呢?

<!-- gh-comment-id:1606774983 --> @yan-rui commented on GitHub (Jun 26, 2023): > > 我现在用的GPU使用的监控工具与nvidia很像。 所以我想,是否需要将nvidia监控工具相关的api替换为现在在用的GPU的监控工具api,就可以用上nvitop了? > > 我认为是可以的 我修改哪些地方可以达到这个目的呢?
Author
Owner

@Junyi-99 commented on GitHub (Jun 26, 2023):

抱歉,你需要自己阅读源代码,找到需要修改的位置

<!-- gh-comment-id:1606780019 --> @Junyi-99 commented on GitHub (Jun 26, 2023): > 抱歉,你需要自己阅读源代码,找到需要修改的位置
Author
Owner

@yan-rui commented on GitHub (Jun 26, 2023):

好的, 多谢回复,我仔细看看代码

<!-- gh-comment-id:1606851908 --> @yan-rui commented on GitHub (Jun 26, 2023): 好的, 多谢回复,我仔细看看代码
Author
Owner

@Junyi-99 commented on GitHub (Mar 11, 2024):

I've successfully adapted nvitop for AMD platforms using ROCm and tested it on mi50, mi100, and mi210 machines without affecting NVIDIA functionality.

I'd like to contribute these changes to extend nvitop's usability to AMD users.

Looking forward to your feedback!

  • MI50:
    mi50

  • MI100:
    mi100

  • MI210:
    mi210

<!-- gh-comment-id:1988448197 --> @Junyi-99 commented on GitHub (Mar 11, 2024): I've successfully adapted nvitop for AMD platforms using ROCm and tested it on `mi50`, `mi100`, and `mi210` machines without affecting NVIDIA functionality. I'd like to contribute these changes to extend nvitop's usability to AMD users. Looking forward to your feedback! - **MI50:** <img width="1024" alt="mi50" src="https://github.com/XuehaiPan/nvitop/assets/14367694/5c8b64ea-6b4d-4232-981a-489183ecc9bd"> - **MI100:** <img width="1024" alt="mi100" src="https://github.com/XuehaiPan/nvitop/assets/14367694/53abab11-9dbe-4d84-b3b3-3f3020fa2071"> - **MI210:** <img width="1024" alt="mi210" src="https://github.com/XuehaiPan/nvitop/assets/14367694/943f7a6f-c60c-454d-a4d5-3a4cbb7707d9">
Author
Owner

@Junyi-99 commented on GitHub (Mar 11, 2024):

🤗 Really looking forward to nvitop's support for AMD GPUs, just like ROCm offers CUDA compatibility!

<!-- gh-comment-id:1988490896 --> @Junyi-99 commented on GitHub (Mar 11, 2024): 🤗 Really looking forward to `nvitop`'s support for AMD GPUs, just like ROCm offers CUDA compatibility!
Author
Owner

@johnnynunez commented on GitHub (Oct 28, 2024):

any news?

<!-- gh-comment-id:2441603832 --> @johnnynunez commented on GitHub (Oct 28, 2024): any news?
Author
Owner

@intexcor commented on GitHub (Jul 3, 2025):

any news?

Use https://github.com/intexcor/gputop

<!-- gh-comment-id:3033409800 --> @intexcor commented on GitHub (Jul 3, 2025): > any news? Use https://github.com/intexcor/gputop
Author
Owner

@yizeyi18 commented on GitHub (Mar 12, 2026):

any news?

<!-- gh-comment-id:4044278701 --> @yizeyi18 commented on GitHub (Mar 12, 2026): any news?
Sign in to join this conversation.
No milestone
No project
No assignees
1 participant
Notifications
Due date
The due date is invalid or out of range. Please use the format "yyyy-mm-dd".

No due date set.

Dependencies

No dependencies set.

Reference: github-starred/nvitop#43
No description provided.