装 GPU 开发环境,有点像自己配一台零件繁多的电脑:源代码、依赖和配置只要有一处对不上,构建就可能失败。AMD 的 TheRock(The HIP Environment and ROCm Kit)想把这件事收拢起来。它是面向 HIP 与 ROCm 的轻量开源构建平台;ROCm 是 AMD 的 GPU 计算软件底座,HIP 则是让开发者编写或迁移 AMD GPU 程序的接口和工具链。
TheRock 的核心是一套 CMake super-project——由一个顶层工程统一协调多个相互依赖的子项目。开发者可以用它从源码构建 HIP、ROCm,以及采用 ROCm 的 PyTorch 和 JAX;项目还提供 ROCm 与 PyTorch 的每日构建版本,覆盖多个 Linux 发行版和原生 Windows,并配有构建、测试和发布流程。它瞄准的痛点很具体:项目说明称,多数构建失败来自操作系统在依赖和项目配置上的细小差异。
不过,TheRock 目前仍处于早期预览阶段。项目也提醒,源码构建虽然更灵活,但在受支持的配置中,直接安装预构建版本通常更快、更省事。