在rk3588上部署并使用NPU/CPU方式运行Deepseek-R1-1.5b/7b模型（全流程实操）

嵌入式郑工

已于 2025-03-11 17:14:48 修改

阅读量1.6k

点赞数 22

文章标签： ubuntu linux ai 机器学习

于 2025-03-04 11:38:12 首次发布

本文链接：https://blog.youkuaiyun.com/JISIC/article/details/145993322

版权

开发环境：

主机环境：windows下vmware中的ubuntu22.04版本

注意：转换7B模型的时候需要用到32G的运行内存，本人实测，建议最少给虚拟机分配40G的运存才能稳定转换。不建议使用swap交换空间，会经常卡死。如果电脑内存不够建议加内存条

1.5b模型转换成rkllm格式后的大小为1.9G，7b模型转换后为7.7G左右，请保证ubuntu下磁盘空间剩余充足

目标机（orangepi5pro开发板）环境：

注意：如果想使用rk3588的npu运行和推理deepseek-r1-1.5b/7b模型,npu版本最低为0.9.8，本人亲测0.9.6版本不支持。进入到开发板的系统中，执行cat /proc/rknpu/version或者cat /sys/kernel/debug/rknpu/version可查看当前的版本。如果不知道如何更换npu版本，请参照香橙派5 RK3588S NPU驱动升级0.9.8_香橙派3588安装github系统linux-优快云博客

部署步骤（以下1-6步都在虚拟机的ubuntu里操作）

1，安装anaconda软件环境，详细可参考在Ubuntu中安装Anaconda和创建虚拟环境（保姆级教学，值得借鉴与信任）_ubuntu anaconda创建虚拟环境-优快云博客

2，anaconda安装完成后，执行conda create -n RKLLM-Toolkit python=3.8新建conda环境
，然后用conda env list命令可验证是否创建成功

conda环境创建成功后，执行conda activate RKLLM-Toolkit，激活并进入到RKLLM-Toolkit环境（退出已激活的conda环境到默认的base的命令为conda deactivate）

下图为正确安装anaconda和正确激活RKLLM-Toolkit环境的示例

3，获取rkllm转换工具

GitHub - airockchip/rknn-llm将此网站的仓库git clone下来，进入到rknn-llm-main/rkllm-toolkit目录下并执行pip install rkllm_toolkit-1.1.4-cp38-cp38-linux_x86_64.whl,耐心等待安装完成

4，下载DeepSeek-R1-Distill-Qwen-1.5B/DeepSeek-R1-Distill-Qwen-7B原始模型

方法一：直接通过浏览器打开并下载deepseek-ai/DeepSeek-R1-Distill-Qwen-7B at main (huggingface.co)

方法二：在ubuntu终端通过命令下载

安装huggingface工具并指定源镜像地址：

pip3 install huggingface-cli -i https://mirrors.huaweicloud.com/repository/pypi/simple

export HF_ENDPOINT=https://hf-mirror.com

将deepseek-ai/DeepSeek-R1-xx模型资源下载到本地当前目录下：

huggingface-cli download deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B --local-dir . --local-dir-use-symlinks False

5，创建必要的文件，保证后面能正常执行转换

克隆的仓库里的rknn-llm-release-v1.1.4/rkllm-toolkit/examples目录下只有一个test.py文件，我们需要在这里新建data_quant.json和export_deepseek_7b_rkllm.py这两个文件，

data_quant.json文件内容示例如下：

[{"input":"Human: 你好！\nAssistant: ", "target": "你好！我是人工智能助手KK！"},{"input":"Human: 你可以干什么？\nAssistant: ", "target": "我可以唱跳rap"}]

export_deepseek_7b_rkllm.py文件内容示例如下：

rom rkllm.api import RKLLM
import os
#os.environ['CUDA_VISIBLE_DEVICES']='0'

modelpath = './huggingface/deepseek-r1-7b'
llm = RKLLM()

# Load model
# Use 'export CUDA_VISIBLE_DEVICES=0' to specify GPU device
# options ['cpu', 'cuda']
ret = llm.load_huggingface(model=modelpath, model_lora = None, device='cpu')
if ret != 0:
print('Load model failed!')
exit(ret)

# Build model
dataset = "./data_quant.json"
qparams = None
ret = llm.build(do_quantization=True, optimization_level=1, quantized_dtype='w8a8',
quantized_algorithm='normal', target_platform='rk3588', num_npu_core=3,
extra_qparams=qparams,dataset=dataset)
if ret != 0:
print('Build model failed!')
exit(ret)

# Export rkllm model
ret = llm.export_rkllm(f"./deepseek-7b-w8a8-rk3588.rkllm")
if ret != 0:
print('Export model failed!')
exit(ret)

6，使用rknn-toolkit工具将原始的huggingface格式的DeepSeek-R1-xx模型转化为瑞芯微支持的rkllm格式，这样才能使用瑞芯微的npu进行推理(注意：需要在conda activate RKLLM-Toolkit环境下运行，请先激活)

执行python export_deepseek_7b_rkllm.py命令，开始转换，整个过程大概要20-30分钟

转换过程如图所示：这是一个相当大的任务，cpu和磁盘以及内存的利用率很高

转换完成后会生成deepseek-7b-w8a8-rk3588.rkllm这个文件，将其传到开发板的系统里

7，部署开发板环境，以便使用rk3588的npu运行转换后的deepseek模型（下面步骤都在开发板的ubuntu系统中操作）

git clone GitHub - airockchip/rknn-llm

sudo cp rkllm-runtime/Linux/librkllm_api/aarch64/librkllmrt.so /usr/lib/librkllmrt.so

cd examples/DeepSeek-R1-Distill-Qwen-1.5B_Demo/deploy

chmod a+x build-linux.sh

./build-linux.sh

执行完上面的脚本后，会在当前文件夹会出现install/demo_Linux_aarch64目录，将第六步生成的转换后的模型拷贝到这里

cd install/demo_Linux_aarch64

diff lib/librkllmrt.so /usr/lib/librkllmrt.so //验证一下刚生成的这个库文件是不是和rknn-llm-main/rkllm-runtime/Linux/librkllm_api/aarch64/librkllmrt.so的一样，要保证它们是1.1.4版本的才能满足这个模型的运行要求