文章目录
1、前言
本文主要介绍如何在ubuntu下的搭建RKNN-ToolKit2开发环境。
2、下载安装Miniconda
Miniconda 是一个轻量级的 Conda 发行版,用于管理 Python 版本和虚拟环境。
2.1、下载miniconda
# 下载 Linux 64 位 Miniconda 最新版安装脚本
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
2.2、安装miniconda
# 安装 Miniconda
bash Miniconda3-latest-Linux-x86_64.sh
安装完成后会自动设置环境变量。打开新的终端,发现用户名前出现(base),就代表安装成功:
3、下载安装RKNN-ToolKit2
RKNN-ToolKit2 开发套件运行在PC (x86_64/arm64)平台上,提供了模型转换、量化功能、模型推理、性能和内存评估、量化精度分析、模型加密等功能。
3.1、创建虚拟环境
# 创建一个名为rknn-toolkit2的虚拟环境,并指定python版本
conda create -n rknn-toolkit2 python=3.8
conda activate rknn-toolkit2
3.2、下载RKNN-ToolKit2
# 可以从官方地址获取:git clone https://github.com/airockchip/rknn-toolkit2.git
# 因为拉取速度太慢,这里使用野火提供的仓库
git clone https://gitee.com/LubanCat/lubancat_ai_manual_code.git
3.3、安装RKNN-ToolKit2安装依赖和包
# 进入rknn_toolkit2目录
cd lubancat_ai_manual_code/dev_env/rknn_toolkit2
# 配置pip源
pip3 config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple/
# 安装依赖库
pip3 install numpy
pip3 install -r doc/requirements_cp38-1.5.0.txt
# 安装rknn_toolkit2
pip3 install packages/rknn_toolkit2-1.5.0+1fa95b5c-cp38-cp38-linux_x86_64.whl
4、验证
(rknn-toolkit2) lubancat@lubancat-vm:~$ python3
Python 3.8.10 (default, May 26 2023, 14:05:08)
[GCC 9.4.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> from rknn.api import RKNN
>>>
5、总结
参考文章:
https://doc.embedfire.com/linux/rk356x/Ai/zh/latest/lubancat_ai/env/toolkit2.html#