RoCm support
A pull request for this has already been merged.
- #490 by @IlyasMoutawwakil — merged
- Dominant language
- Python
- Stars
- 1.9k
- Forks
- 323
- Avg merge
- 1d 12h
- Merged PRs (30d)
- 12
Description
- CodeCarbon version: main
- Python version: all
- Operating System: Linux
Description
It would be great to add RoCm support to codecarbon. For now, codecarbon requires and depends on pynvml, that is an interface only for Nvidia GPUs. However, using AMD GPUs is becoming easier and easier, and we can expect some researchers to make the switch from Nvidia.
A WIP alternative for RoCm seem to be https://github.com/RadeonOpenCompute/pyrsmi
I could possibly submit a PR if there is interest.
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Start by reviewing CodeCarbon’s current pynvml dependency and the closed linked pull request #490. Evaluate the proposed pyrsmi project as a ROCm path; done means CodeCarbon supports AMD GPUs on Linux rather than relying only on the NVIDIA-specific interface.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python
- Domain
- tooling
- Issue type
- Feature
- Difficulty
- 5/5
- Estimated time
- Over a week
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 25/100