使用瑞芯微的Dockerfile创建模型krnn模型导出与交叉编译环境
·
最近遇到了奇怪的事情,之前配制好的rknn环境丢失了,重新配置又遇到了各种问题
库版本不兼容whl文件安装报错等,由于手动创建环境十分麻烦,经常会出现各种库版本不匹配的问题。因此,今天考虑从Dockerfile入手创建docker环境,使得官方提供的完善环境直接可用。


以全国产化开发板瑞芯微为例,首先下载官方提供的Dockerfile
https://github.com/airockchip/rknn-toolkit2/tree/master/rknn-toolkit2/docker/docker_file/ubuntu_20_04_cp38
我自己也留了一个备份,以备不时之需
https://gitee.com/liuzhibang123/rknn_docker_file.git(支持国产,嘿嘿)
然后,使用如下命令创建docker image
docker build -f Dockerfile_ubuntu_20_04_for_cp38.txt -t rknn:20.04 .

创建完成后,可以使用docker images命令查看到新创建的镜像

删除之前配置失败的docker容器,然后基于docker file生成的新镜像重新创建容器
docker rm lzb_rknn
docker run -d --name lzb_rknn -p 5778:22 -it --network bridge --gpus all --restart unless-stopped -w /workdir --privileged rknn:20.04
docker start lzb_rknn
docker exec -it lzb_rknn bash

然后,尝试导出rknn模型,交叉编译模型推理代码
更多推荐
所有评论(0)