Add ROCm support for AMD Instinct GPUs
维护者通常 1 天内回复
评估
- 难度
- 5/5
- 预计耗时
- 一周以上
- 新手友好度
- 20/100
- Issue 类型
- 功能
- 描述清晰度
- 基本清楚
- 活跃度
- 停滞
调研方向
Start by reading the build logic that detects the GPU toolkit and the existing Dockerfile; the issue does not name their paths. The open pull request #9153 already addresses this work, so coordinate there rather than starting a separate implementation. Done would mean ROCm toolkit detection builds the C++ extensions and an official ROCm container recipe is provided.
由索引模型根据 Issue 内容生成。
描述
Is your feature request related to a problem?
MONAI doesn't run on AMD Instinct GPUs out of the box.
The bulk of the library is already portable and needs no changes, with the following exceptions:
- The compiled C++ extensions are silently skipped. The build detects the toolkit by a vendor-specific path that isn't set on ROCm, so it concludes there's no GPU toolchain available. The build then reports success with the extensions missing, and the failure only surfaces much later as a missing-op error at runtime.
- There's no ROCm container recipe. The existing Dockerfile is built on a vendor-specific base image, so there's no supported starting point on this platform.
Describe the solution you'd like
First-class ROCm support for AMD Instinct GPUs:
- Toolkit detection that recognises a ROCm installation, so the compiled extensions build instead of being skipped.
- An official ROCm container recipe alongside the existing Dockerfile.
We'd like this to be minimal and non-invasive — no abstraction layer and no per-vendor branching in model or transform code.
Describe alternatives you've considered
N/A
Additional context
Target platform: AMD Instinct MI300X / MI325X (gfx942) and MI355X (gfx950), ROCm 10.0, Ubuntu 24.04, Python 3.12.
- 主要语言
- Python
- 星标
- 8.7k
- 派生
- 1.6k
- 平均合并
- 3 天 10 小时
- 30 天内合并 PR
- 13
环境准备
- 提供 Dockerfile 或 Docker Compose 文件
- 有 Pull Request 模板
- 阅读贡献指南
从这里开始
- 先读完整个 Issue，再读项目的贡献指南。
- 在 Issue 下留言说明你要接手 —— 这能避免两个人做同样的事。
- Fork 仓库，在一个分支上完成修改。
- 提交 Pull Request，并在描述里引用这个 Issue 编号。
Project-MONAI/MONAI 的其他 Issue
-
ProbMapProducer raises IndexError on numpy 2.x: SlidingPatchWSIDataset emits float locations可能已有人在做 @nilapate 于 9 天前认领。 未关闭
难度 2/5 1-3 小时 新手友好度 82/100
Project-MONAI/MONAI#9135 ·
维护者通常 1 天内回复
-
RandGridPatch crashes for documented 0 or None patch_size entries可能已有人在做 @Rajioba1 于 66 天前认领。 未关闭
难度 2/5 1-3 小时 新手友好度 78/100
Project-MONAI/MONAI#9046 ·
维护者通常 1 天内回复
-
Incorrect type hint in `pad_func`可能已有人在做 @Harshal875 于 71 天前认领。 未关闭
难度 1/5 1 小时以内 新手友好度 88/100
Project-MONAI/MONAI#9026 ·
维护者通常 1 天内回复
-
难度 3/5 1-2 天 新手友好度 74/100
Project-MONAI/MONAI#9151 ·
维护者通常 1 天内回复
-
PR Update Action可能已有人在做 @KumoLiu 于 8 天前认领。 未关闭
Project-MONAI/MONAI#9141 · 已指派 3 人 ·
维护者通常 1 天内回复
查看 Project-MONAI/MONAI 的全部 Issue
相似的 Issue
-
automated issue report
难度 1/5 1 小时以内 新手友好度 85/100
RapidAI/RapidOCRDocs#119 ·
-
难度 2/5 1-3 小时 新手友好度 70/100
-
难度 2/5 1-3 小时 新手友好度 85/100
btclib-org/btclib-node#1833 ·
维护者通常 1 天内回复
-
IRIS reader: no-data velocity bins (DB_VEL, DB_VELC) returned as 0.0 m/s instead of NaN可能已有人在做 @syedhamidali 今天认领。 未关闭
难度 2/5 1-3 小时 新手友好度 72/100
维护者通常 2 天内回复
-
难度 1/5 1 小时以内 新手友好度 80/100
elodin-sys/elodin#890 ·
维护者通常 1 天内回复