首先安装好conda,我是用的是anaconda,miniconda也可以。
下载rknn_toolkit的轮子。可以直接在瑞芯微的git仓库中下载,地址为:github.com/rockchip-linux/rknn-toolkit/releases。我这里下载的是1.7.5版本的。选择rknn-toolkit-v1.7.5-packages.zip下载,之后解压,pip install rknn_toolkit-1.7.5-cp36-cp36m-win_amd64.whl。注意只支持python3.6的版本,如果下载不下来可以去gitee.com/FIRC/pythonlibs_whl_mirror搜索下载
requirements-cpu.txt
tensorflow==1.14.0
torch==1.10.0
torchvision==0.11.0
mxnet==1.5.0
requirements-cpu-ubuntu20.04_py38.txt
tensorflow==2.2.0
torch==1.10.0
torchvision==0.11.0
mxnet==1.5.0
requirements-gpu.txt
tensorflow-gpu==1.14.0
torch==1.10.0
torchvision==0.11.0
mxnet-cu101==1.5.0
打开anaconda prompt命令行,
conda create -n rknn python=3.6
启动conda环境。
conda activate rknn
从本地安装torch
pip install torch==1.10.0+cpu torchvision==0.11.0+cpu --find-links=你的路径
安装mxnet,rknn-toolkit官方要求的版本是mxnet==1.5.0
pip install mxnet==1.5.0
安装tenserflow,rknn-toolkit官方要求的版本是tensorflow==1.14.0
pip install tensorflow==1.14.0
安装opencv-python
pip install opencv-python==4.5.4.60
从本地安装rknn-toolkit
pip install rknn_toolkit-1.7.5-cp36-cp36m-win_amd64.whl --find-links=你的路径