最近遇到了奇怪的事情,之前配制好的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模型,交叉编译模型推理代码

更多推荐