跳到主要内容

openRuyi AI 栈概览

openRuyi 提供了一套用于大模型推理的 AI 软件栈,包括 llama.cpp、PyTorch、Triton 和 vLLM,以及 AMD GPU 计算平台 ROCm,开箱即用。

具体使用示例请参阅 llama.cpp 快速入门。

组件​

部分软件提供 CPU 和 ROCm 两个版本。CPU 版本可以在任意 openRuyi riscv64 机器上使用,ROCm 版本需要支持的 AMD 显卡。

软件CPU 版本ROCm 版本说明
llama.cppllama-cppllama-cpp-rocmGGUF 格式推理引擎
PyTorchpython-torchpython-torch-rocm深度学习框架
Tritonpython-tritonGPU 算子编译器
vLLMpython-vllmpython-vllm-rocm高性能推理引擎
ROCmrocr-runtime、rocm-hip、rocblas、rocminfo、rocm-smi 等AMD GPU 运行时、数学库与工具

安装 ROCm 版本的软件包时,所需的 ROCm 运行时和数学库会作为依赖一并安装。

同一软件的 CPU 版本和 ROCm 版本不能同时安装。如需从一个版本切换到另一个版本,请在安装时加上 --allowerasing,例如:

sudo dnf install --allowerasing llama-cpp-rocm

支持的显卡​

openRuyi 的 ROCm 软件包为以下 GPU 架构构建:

GPU 架构显卡
gfx1100 gfx1101RDNA3 系列显卡,例如 Radeon RX 7900 XTX
gfx1200 gfx1201RDNA4 系列显卡,例如 Radeon RX 9070 XT

AMD Instinct 计算卡暂未支持。