RKNN Toolkit2是用來將onnx模型轉成rknn專用模型,并可通過RKNN Toolkit Lite2或者RKNPU調用NPU進行加速計算的工具。
一開始我安裝很多次都無法成功安裝。后來跟售后技術對接,必須是PC平臺的Linux環境才可以。我的電腦是windows,所以我需要用WSL來進行安裝。
打開PowerShell,先給WSL安裝一個ubuntu環境,并安裝好miniconda
wsl --install -d Ubuntu
wsl -d Ubuntu
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
bash ./Miniconda3-latest-Linux-x86_64.sh
創建python3.10環境
conda create -n py310 python=3.10
conda activate py310
拉一下官方庫
git clone https://gitee.com/LubanCat/lubancat_ai_manual_code.git
cd lubancat_ai_manual_code/dev_env/rknn_toolkit2
先安裝rknn toolkit2的依賴,再安裝rknn toolkit2本體,注意選擇對應的python版本文件
pip install -r doc/requirements_cp310-1.5.0.txt
pip install packages/rknn_toolkit2-1.5.0+1fa95b5c-cp310-cp310-linux_x86_64.whl
確認一下是否安裝成功
python
Python 3.10.16 (main, Dec 11 2024, 16:24:50) [GCC 11.2.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> from rknn.api import RKNN
>>>