1. 安装 Anaconda + 配置Python基础环境
1.1 去官网下载并安装Anaconda
1.2 新建并激活一个 python3.6 环境:
conda create --name=rknn python=3.6.8
conda activate rknn
进入虚拟环境后,再执行以下命令安装深度学习框架,如Tensorflow,Pytorch,Keras等。
2. 配置 python 环境
2.1 官方给出的依赖
pip install tensorflow==1.14.0
pip install torch==1.5.1
pip install torchvision==0.4.0
pip install mxnet==1.5.0
2.2 安装 tensorflow==1.14.0 :
pip install tensorflow==1.14.0
2.3 安装 torch
去官网的安装包库下载安装包,并安装:
pip install "D:\SDK\external\rknn-toolkit\packages\torch-1.5.1+cpu-cp36-cp36m-win_amd64.whl"
2.4 安装 torchvision
去官网的安装包库下载安装包,并安装:
pip install "D:\SDK\external\rknn-toolkit\packages\torchvision-0.4.0+cpu-cp36-cp36m-win_amd64.whl"
2.5 安装 mxnet==1.5.0:
pip install mxnet==1.5.0
2.6 安装opencv-python 及 报错的解决方法:
2.6.1 安装opencv:
pip install opencv-python
2.6.2 python3.6安装opencv遇到报错,如下所示:
********************************************************************************
----------------------------------------
ERROR: Failed building wheel for opencv-python
Failed to build opencv-python
ERROR: Could not build wheels for opencv-python which use PEP 517 and cannot be installed directly
2.6.3 解决方法:
原因是使用pip install opencv-python
命令安装的是最新版本,python3.6不支持。所以找一个python3.6支持的版本。如opencv-python==4.3.0.38
。
pip install -i https://pypi.douban.com/simple/ pip install opencv-python==4.3.0.38
2.7 安装 gluoncv :
pip install gluoncv
注:gluoncv 在运行 examples/mxnet 中的例子时会用到。
2.8 安装rknn-toolkit-1.7.1 及 报错:
2.8.1 安装rknn-toolkit-1.7.1
复制出rknn安装包"rknn_toolkit-1.7.1-cp36-cp36m-win_amd64.whl"
,位置在SDK/external/rknn-toolkit/packages
/下,后进入你的packages
/执行以下命令进行安装。
pip install rknn_toolkit-1.7.1-cp36-cp36m-win_amd64.whl
2.8.2 报错:
ERROR: pip's dependency resolver does not currently take into account all the packages that are installed. This behaviour is the source of the following dependency conflicts.
mxnet 1.5.0 requires requests<2.19.0,>=2.18.4, but you have requests 2.22.0 which is incompatible.
matplotlib 3.3.4 requires pillow>=6.2.0, but you have pillow 5.3.0 which is incompatible.
由于 mxnet
所依赖的 requests
和 rknn-toolkit
所依赖的 requests
冲突。
2.8.3 解决方法:
发现mxnet-1.0.0
既不会更换 numpy
也不会更换 requests
,从而导致与 rknn-toolkit
依赖包冲突。rknn-toolkit-1.7.1
也没有依赖报错,更换代码如下:
pip uninstall mxnet
pip install mxnet==1.0.0
2.9 检测rknn是否安装成功,输入以下命令:
python
from rknn.api import RKNN
若无报错则安装RKNN成功,如下:
(rknn) D:\SDK\external\rknn-toolkit\packages> python
Python 3.6.8 |Anaconda, Inc.| (default, Feb 21 2019, 18:30:04) [MSC v.1916 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> from rknn.api import RKNN
>>>