返回卡片页面

RK3568 RKNN-Toolkit2 环境安装教程

在Ubuntu系统上安装和配置RKNN-Toolkit2开发环境

目录

系统要求

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

1. Miniconda安装与配置

1 下载Miniconda

Miniconda是Anaconda的轻量级版本,只包含conda、Python和必要的依赖包。您可以从以下地址下载:

工具下载地址

或直接从官网下载:Miniconda官网

下载Linux版本的安装文件:Miniconda3-latest-Linux-x86_64.sh

2 上传到Ubuntu系统

将下载的安装文件上传到您的Ubuntu系统。如果您在Windows/Mac上,可以使用SCP命令:

在主机上执行(Windows/Mac终端)
# 将文件上传到Ubuntu系统
scp -r ./Miniconda3-latest-Linux-x86_64.sh alientek@192.168.2.193:/home/alientek

注意:请将192.168.2.193替换为您的Ubuntu系统IP地址,将alientek替换为您的用户名。

3 安装Miniconda

在Ubuntu系统上执行安装脚本:

在Ubuntu终端执行
bash Miniconda3-latest-Linux-x86_64.sh

安装过程中会提示:

  1. Enter键查看许可协议
  2. 空格键滚动查看协议内容
  3. 输入yes并回车同意许可协议
  4. 选择安装路径(默认按回车即可)
  5. 输入yes初始化Miniconda3

安装完成后会显示:Thank you for installing Miniconda3!

4 配置Miniconda

重启Ubuntu终端,您会看到前面多了个(base),这表示已进入conda基础环境。

在Ubuntu终端执行
# 退出conda环境(返回系统默认环境)
conda deactivate

# 设置不自动激活conda基础环境(可选)
conda config --set auto_activate false

注意:原教程中的conda config --set auto_activate_base false命令有误,新版本正确命令是conda config --set auto_activate false

5 配置国内镜像源

为了加速软件包下载,建议配置国内镜像源:

配置conda清华源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/

# 配置pip清华源
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
查看已配置的源
conda config --show channels

2. 创建RKNN-Toolkit2环境

1 获取RKNN-Toolkit2源代码

首先,进入您的工作目录并克隆RKNN-Toolkit2源代码:

在Ubuntu终端执行
# 进入工作目录(可根据需要修改)
cd /home/alientek/work/rknn

# 克隆RKNN-Toolkit2源代码
git clone https://github.com/airockchip/rknn-toolkit2.git

提示:注意老版本已弃用:https://gitee.com/rockchip-linux/rknn-toolkit2.git

2 创建Conda环境

为RKNN-Toolkit2创建一个独立的Python环境:

在Ubuntu终端执行
# 创建名为yolo11的Python 3.8环境
conda create --name yolo11 python=3.8

# 当提示"Proceed ([y]/n)?"时,输入y并按回车

# 激活新创建的环境
conda activate yolo11

环境命名:环境名称yolo11是示例名称,您可以根据需要更改为其他名称,如rknn等。

3. 安装RKNN-Toolkit2

1 安装依赖库

进入RKNN-Toolkit2目录,安装必要的依赖库:

在Ubuntu终端执行(确保在yolo11环境中)
# 进入RKNN-Toolkit2目录
cd /home/alientek/work/rknn/rknn-toolkit2/rknn-toolkit2/packages/x86_64

# 安装依赖库(根据Python版本选择对应文件)
pip install -r requirements_cp38-2.3.2.txt

安装成功后,会显示:Successfully installed ...

2 安装RKNN-Toolkit2

安装RKNN-Toolkit2工具包:

在Ubuntu终端执行(确保在yolo11环境中)
pip install rknn_toolkit2-2.3.2-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl

安装成功后,会显示:Successfully installed rknn-toolkit2

注意:文件名可能因版本不同而有所变化,请根据实际文件名进行调整。

4. 测试与验证

1 运行测试代码

进入示例目录,运行测试代码验证安装是否成功:

在Ubuntu终端执行
# 进入示例目录
cd /home/alientek/work/rknn/rknn-toolkit2/rknn-toolkit2/examples/tflite/mobilenet_v1

# 激活环境(如果尚未激活)
conda activate yolo11

# 运行测试代码
python test.py

2 处理常见错误

如果运行测试时出现TensorFlow相关错误:

错误信息:

I rknn-toolkit2 version: 2.3.2
--> Config model
done
--> Loading model
W load_tflite: 'import tensorflow' failed! Please use the following command to reinstall it:
W load_tflite:   pip3 install 'tensorflow>=1.12.0,<=2.14.0'
W load_tflite: In addition, it is recommended that the TensorFlow version is consistent with the version of
the tflite model to avoid the model cannot be parsed.
Load model failed!

请安装正确版本的TensorFlow:

安装TensorFlow
pip3 install 'tensorflow>=1.12.0,<=2.14.0'

# 重新运行测试
python test.py

3 验证安装成功

如果安装成功,您将看到类似以下的输出:

成功输出示例:

----TOP 5-----
[ 156] score:0.928223 class:"Shih-Tzu"
[ 155] score:0.063171 class:"Pekinese, Pekingese, Peke"
[ 205] score:0.004299 class:"Lhasa, Lhasa apso"
[ 284] score:0.003096 class:"Persian cat"
[ 285] score:0.000171 class:"Siamese cat, Siamese"
done

恭喜! RKNN-Toolkit2环境已成功安装并可以正常工作。

5. 常见问题解决

问题1:conda命令找不到

解决方法:重新加载bash配置或重启终端

source ~/.bashrc

问题2:pip安装缓慢或失败

解决方法:检查是否已配置国内源,或临时使用镜像源

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple 包名

问题3:Git克隆速度慢

解决方法:使用Gitee镜像或配置Git代理

# 使用Gitee镜像
git clone https://gitee.com/rockchip/rknn-toolkit2.git

问题4:Python版本不兼容

解决方法:确保使用Python 3.8版本,这是RKNN-Toolkit2 2.3.2的推荐版本

# 检查Python版本
python --version

# 如果不是3.8.x,请重新创建conda环境
conda create --name rknn python=3.8

环境管理常用命令

Conda环境管理命令
# 查看所有环境
conda env list

# 激活环境
conda activate 环境名

# 退出当前环境
conda deactivate

# 删除环境
conda remove --name 环境名 --all

# 查看已安装的包
conda list
RKNN-Toolkit2验证命令
# 验证RKNN-Toolkit2安装
python -c "from rknn.api import RKNN; print('RKNN-Toolkit2导入成功')"

# 查看RKNN-Toolkit2版本
python -c "from rknn.api import RKNN; rknn = RKNN(); print(rknn.version)"