1. 环境准备:为什么选择Docker?
如果你正在玩瑞芯微(Rockchip)的AI开发板,比如我手头的AIO-3399ProC,那你大概率绕不开一个东西:RKNN Toolkit。这是官方提供的模型转换和推理工具链,说白了,就是把你在PC上训练好的AI模型(比如TensorFlow、PyTorch的),转换成能在开发板NPU上高效运行的格式。
听起来很简单对吧?但实际操作起来,新手(甚至老手)最容易栽跟头的就是环境配置。我刚开始折腾的时候,照着官方文档一步步装Python、装各种依赖包,结果不是这个包版本冲突,就是那个库死活装不上,光一个tensorflow==1.14.0的兼容性问题就耗了我大半天。那种感觉,就像拼乐高缺了一块关键零件,非常抓狂。
后来我发现,官方其实提供了一个“懒人包”——预配置好的Docker镜像。这个镜像里,Python、RKNN Toolkit、TensorFlow、PyTorch、OpenCV等等所有需要的软件,版本都给你配好了,而且是经过官方测试完全兼容的。这就好比你去餐厅,不用自己买菜、洗菜、研究菜谱,直接给你上一盘色香味俱全的成品菜。
所以,在Ubuntu PC端用Docker部署RKNN环境,核心优势就三个字:省事、干净、可复现。
- 省事:跳过所有依赖地狱,一键导入就能用。
- 干净:所有环境装在容器里,不污染你主机系统。今天用RKNN 1.6.0,明天想测试1.4.0,再起一个容器就行,互不干扰。
- 可复现:团队协作时,大家用同一个镜像,保证环境完全一致,避免“在我机器上是好的”这种问题。
在开始动手前,我们先明确一下这次部署的基础环境,这也是我实测下来最稳的组合:
| 组件 | 推荐版本 | 说明 |
|---|---|---|
| PC操作系统 | Ubuntu 18.04 LTS 或 20.04 LTS | 官方Docker镜像主要针对这两个长期支持版测试,兼容性最好。我用的是18.04。 |
| RKNN Toolkit 版本 | 1.6.0 / 1.4.0 / 1.3.0 | 根据你的开发板NPU型号和需求选择。新板子建议用1.6.0。 |
| Docker 镜像 | rknn-toolkit-1.x.x-docker.tar.gz |
从官方资源或社区获取,对应你选择的RKNN版本。 |
| Python | 3.6 | 注意:官方镜像内置的是Python 3.6。虽然主机装3.8也行,但容器内是3.6,这点要知晓。 |
准备好这些,我们就可以开始了。整个过程就像搭积木,一步一步来,非常清晰。
2. 安装Docker:打好地基
Docker是我们整个环境的“集装箱码头”,必须先把它搭建好。官方的安装指南很详细,但有些步骤对于国内网络环境可能需要“变通”一下。我这里结合自己的踩坑经验,给你梳理一个更顺畅的流程。
首先,卸载可能存在的旧版本。这一步很重要,避免后续冲突。
sudo apt-get remove docker docker-engine docker.io containerd runc
系统可能会提示有些包没安装,不用管,继续就行。
接下来,我们采用官方推荐的通过仓库安装的方式。这样以后更新管理都方便。
第一步,安装

4470

被折叠的 条评论
为什么被折叠?



