首页
/ TheRock 的安装和配置教程

TheRock 的安装和配置教程

2025-05-23 00:01:08作者:温艾琴Wonderful

1. 项目基础介绍和主要编程语言

TheRock 是一个轻量级开源构建系统,主要用于 HIP (High-performance Isotropic Particles) 和 ROCm (Radeon Open Compute) 的源码构建。它提供了一个 CMake 超级项目,可以用来构建 HIP 和 ROCm 的源代码,并且支持构建 PyTorch 与 ROCm 的源码整合。目前,TheRock 项目处于早期预览状态,但正在积极开发中,并欢迎贡献者参与。该项目的主要编程语言是 C++,同时也涉及到一些 Python 脚本用于构建和配置。

2. 项目使用的关键技术和框架

TheRock 使用了以下关键技术和框架:

  • CMake:一个跨平台的安装(编译)工具,能够使用简单的声明性语句描述所有平台的安装(编译过程)。
  • HIP:一个由 AMD 开发的跨平台的高性能计算编程模型,它允许开发者用类 CUDA 的语言来编写代码,可以在 AMD GPU 和 NVIDIA GPU 上运行。
  • ROCm:AMD 的开源高性能计算平台,提供了一套完整的工具和库,以支持高性能计算(HPC)和深度学习应用。

3. 项目安装和配置的准备工作及详细步骤

准备工作

在开始安装 TheRock 之前,请确保您的系统满足以下要求:

  • 操作系统:支持 Ubuntu 24.04 或 Windows 11 (VS 2022)。
  • Git 版本控制系统,用于克隆和更新项目代码。
  • Python 3,用于运行项目中的脚本。

安装步骤

Ubuntu (24.04) 系统

  1. 安装必要的依赖项:

    sudo apt install gfortran git git-lfs ninja-build cmake g++ pkg-config xxd patchelf automake python3-venv python3-dev libegl1-mesa-dev
    
  2. 克隆 TheRock 仓库:

    git clone https://github.com/ROCm/TheRock.git
    cd TheRock
    
  3. 初始化 Python 虚拟环境并安装 Python 依赖:

    python3 -m venv .venv
    source .venv/bin/activate
    pip install -r requirements.txt
    
  4. 下载子模块并应用补丁:

    python ./build_tools/fetch_sources.py
    

Windows 11 系统

  1. 遵循 Windows 支持指南安装所需工具。

  2. 克隆 AMDGPU Windows 互操作库:

    git clone https://github.com/nod-ai/amdgpu-windows-interop.git
    
  3. 克隆 TheRock 仓库:

    git clone https://github.com/ROCm/TheRock.git
    cd TheRock
    
  4. 初始化 Python 虚拟环境并安装 Python 依赖:

    python -m venv .venv
    .venv\Scripts\Activate.bat
    pip install -r requirements.txt
    
  5. 下载子模块并应用补丁:

    python ./build_tools/fetch_sources.py
    

安装完成后,您可以根据需要自定义构建配置,并使用 CMake 开始构建 ROCm/HIP:

cmake -B build -GNinja .

请根据 CMake 的输出信息和项目文档进行进一步的配置和构建。

登录后查看全文
热门项目推荐