RK3568 NPU :RKNN-ToolKit2环境搭建
1. 安装Miniconda3
下载 Linux 64 位 Miniconda 最新版安装脚本
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
安装 Miniconda
bash Miniconda3-latest-Linux-x86_64.sh -u
安装完成后会自动设置环境变量。打开新的终端,发现用户名前出现(base),就代表安装成功:
如果以前安装过conda,建议先删除,再安装,避免出现意外情况。
RKNN-ToolKit2 开发套件运行在PC (x86_64/arm64)平台上,提供了模型转换、量化功能、模型推理、性能和内存评估、量化精度分析、模型加密等功能。
If you'd prefer that conda's base environment not be activated on startup,run the following command when conda is activated:conda config --set auto_activate_base falseYou can undo this by running `conda init --reverse $SHELL`? [yes|no]
[no]>>> no
创建一个名为rknn-toolkit2的虚拟环境,并指定python版本
conda create -n rktoolkit python=3.8
创建过程中会有提示信息,一路接受
即可。
切换到虚拟环境rktoolkit
conda activate rktoolkit
2. 安装RKNN-Toolkit2
下载RKNN-ToolKit2
git clone https://ghproxy.cn/https://github.com/airockchip/rknn-toolkit2.git
安装RKNN-ToolKit2安装依赖和包
进入rknn_toolkit2目录
cd rknn-toolkit2
配置pip源
pip3 config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple/
安装依赖库
注意:每个版本的rknntoolkit的路径会有点不一样,需要自行到目录中找到依赖包requirements_cp38-x.x.x.txt
和工具包whlrknn_toolkit2-xxx.whl
。
pip3 install numpy
pip3 install -r packages/x86_64/requirements_cp38-2.3.2.txt
安装rknn_toolkit2
pip3 install packages/x86_64/rknn_toolkit2-2.3.2-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
验证
python3
rktoolkit) meanshe@aic:/opt/rknn-toolkit2/rknn-toolkit2$ python3
Python 3.8.20 (default, Oct 3 2024, 15:24:27)
[GCC 11.2.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> from rknn.api import rknn
>>>
如果没有报错,就表示安装正确。
附录:清除miniconda3
删除 Miniconda3 目录
直接删除 Miniconda3 的安装目录,通常位于用户主目录下(如 ~/miniconda3
或 /opt/miniconda3
)。使用以下命令:
rm -rf ~/miniconda3
如果是系统级安装,可能需要 root 权限:
sudo rm -rf /opt/miniconda3
清理环境变量
检查并清理 shell 配置文件(如 ~/.bashrc
、~/.bash_profile
、~/.zshrc
)中与 Conda 相关的环境变量。使用文本编辑器打开这些文件,删除以下类似内容:
# >>> conda initialize >>>
# !! Contents within this block are managed by 'conda init' !!
...
# <<< conda initialize <<<
移除隐藏的 Conda 配置
删除 Conda 的隐藏配置文件(位于用户主目录下):
rm -rf ~/.conda
rm -rf ~/.condarc
检查 PATH 变量
确保 Conda 的路径不再出现在 PATH
环境变量中。运行以下命令检查:
echo $PATH
如果仍有 Conda 路径(如 ~/miniconda3/bin
),需手动从 PATH
中移除。
验证卸载
重新打开终端或运行以下命令使更改生效:
source ~/.bashrc # 或其他 shell 配置文件
验证 Conda 是否已彻底删除:
which conda
如果无输出,则说明已成功删除。