TorchMorph:PyTorch 生态的 CUDA 加速形态学库
kornia_foss · x · 2026-08-26
Kai Zhao 发布了 TorchMorph,这是一个基于 PyTorch 的新形态学变换库。它针对 GPU 训练循环进行了优化,通过融合 CUDA 内核直接在 CUDA 张量上操作。与 CPU 版本的 SciPy 相比,批处理执行在灰度形态学上吞吐量最高提升 1100 倍,精确欧几里得距离变换最高提升 350 倍。API 设计与 SciPy 高度兼容,便于现有管线迁移。
所属事件:TorchMorph 发布:PyTorch 生态的 CUDA 加速形态学库(2 条相关)→
「Infra」频道最新
- ASML CFO 称美国限制对华 DUV 销售将适得其反 — pstAsiatech · 2026-10-15
- ASML 业绩超预期仍跌 2%,订单排至 2029 年后 — TiernanRayTech · 2026-10-15
- Bloomberg 算账:得州 AI 数据中心 1GW 场地建设成本失控 — rohanpaul_ai · 2026-10-15
- 系统设计面试必懂的 7 个分布式算法 — goyalshaliniuk · 2026-10-15
- 面壁 MiniCPM-V 4.7 端侧跑 35B 多模态,仅 3GB 内存、每 token 激活 3B — nevrekaraishwa2 · 2026-10-15
- Sipeed MaixStation 产品页上线,端侧 AI 开发套件来了 — xlltt · 2026-10-15