在Windows环境下的rknn-toolkit环境搭建

2024-05-14 22:28

本文主要是介绍在Windows环境下的rknn-toolkit环境搭建,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

在Windows环境下的rknn-toolkit环境搭建

最近需要使用RK3566来跑YOLOv5。想要实现目标就需要使用rknn-toolkit工具将模型转化成可以运行在rk3588的NPU上的模型。只是转换模型的话,没有必要使用GPU,所以我这里安装的是CPU版的,所需的组件按照requirements-cpu.txt中的要求安装。

一、准备工作

  1. 安装好conda,我是用的是anaconda,miniconda应该也可以。
  2. 下载rknn_toolkit的轮子。可以直接在瑞芯微的git仓库中下载,地址为:https://github.com/rockchip-linux/rknn-toolkit/releases。我这里下载的是1.7.1版本的。没试过1.7.3版本,1.7.5版本就不支持RK3566了。选择rknn-toolkit-v1.7.1-packages.zip下载,之后解压,将里面的rknn_toolkit-1.7.1-cp36-cp36m-win_amd64.whl复制到conda目录下的pkgs目录里面。
  3. 下载torch和torchvision。rknn-toolkit官方要求的版本是torch==1.5.1和torchvision==0.4.0。可以直接在pytorch官方目录下下载,地址为:https://download.pytorch.org/whl/torch_stable.html。我这里下载的是torch-1.6.0+cpu-cp36-cp36m-win_amd64.whl和torchvision-0.7.0+cpu-cp36-cp36m-win_amd64.whl。这两个轮子也要放到pkgs目录下面。

二、搭建环境

  1. 打开cmd命令行,并且切换工作目录到conda下的pkgs目录。
  2. 创建conda环境。
    conda create -n rknn-toolkit-1.7.1 python=3.6.8
  3. 启动conda环境。
    conda activate rknn-toolkit-1.7.1
  4. pip换源
python -m pip install --upgrade pip
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
  1. 从本地安装torch
    pip install torch==1.6.0+cpu torchvision==0.7.0+cpu --find-links=你的路径
  2. 安装mxnet,rknn-toolkit官方要求的版本是mxnet==1.5.0
    pip install mxnet==1.5.0
  3. 安装tenserflow,rknn-toolkit官方要求的版本是tensorflow==1.14.0
    pip install tensorflow==1.14.0
  4. 安装opencv-python
    pip install opencv-python==4.5.4.60
  5. 从本地安装rknn-toolkit
    pip install rknn_toolkit-1.7.1-cp36-cp36m-win_amd64.whl --find-links=你的路径

这篇关于在Windows环境下的rknn-toolkit环境搭建的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/990022

相关文章

阿里开源语音识别SenseVoiceWindows环境部署

SenseVoice介绍 SenseVoice 专注于高精度多语言语音识别、情感辨识和音频事件检测多语言识别: 采用超过 40 万小时数据训练,支持超过 50 种语言,识别效果上优于 Whisper 模型。富文本识别:具备优秀的情感识别,能够在测试数据上达到和超过目前最佳情感识别模型的效果。支持声音事件检测能力,支持音乐、掌声、笑声、哭声、咳嗽、喷嚏等多种常见人机交互事件进行检测。高效推

搭建Kafka+zookeeper集群调度

前言 硬件环境 172.18.0.5        kafkazk1        Kafka+zookeeper                Kafka Broker集群 172.18.0.6        kafkazk2        Kafka+zookeeper                Kafka Broker集群 172.18.0.7        kafkazk3

安装nodejs环境

本文介绍了如何通过nvm(NodeVersionManager)安装和管理Node.js及npm的不同版本,包括下载安装脚本、检查版本并安装特定版本的方法。 1、安装nvm curl -o- https://raw.githubusercontent.com/nvm-sh/nvm/v0.39.0/install.sh | bash 2、查看nvm版本 nvm --version 3、安装

【IPV6从入门到起飞】5-1 IPV6+Home Assistant(搭建基本环境)

【IPV6从入门到起飞】5-1 IPV6+Home Assistant #搭建基本环境 1 背景2 docker下载 hass3 创建容器4 浏览器访问 hass5 手机APP远程访问hass6 更多玩法 1 背景 既然电脑可以IPV6入站,手机流量可以访问IPV6网络的服务,为什么不在电脑搭建Home Assistant(hass),来控制你的设备呢?@智能家居 @万物互联

高并发环境中保持幂等性

在高并发环境中保持幂等性是一项重要的挑战。幂等性指的是无论操作执行多少次,其效果都是相同的。确保操作的幂等性可以避免重复执行带来的副作用。以下是一些保持幂等性的常用方法: 唯一标识符: 请求唯一标识:在每次请求中引入唯一标识符(如 UUID 或者生成的唯一 ID),在处理请求时,系统可以检查这个标识符是否已经处理过,如果是,则忽略重复请求。幂等键(Idempotency Key):客户端在每次

AI Toolkit + H100 GPU,一小时内微调最新热门文生图模型 FLUX

上个月,FLUX 席卷了互联网,这并非没有原因。他们声称优于 DALLE 3、Ideogram 和 Stable Diffusion 3 等模型,而这一点已被证明是有依据的。随着越来越多的流行图像生成工具(如 Stable Diffusion Web UI Forge 和 ComyUI)开始支持这些模型,FLUX 在 Stable Diffusion 领域的扩展将会持续下去。 自 FLU

pico2 开发环境搭建-基于ubuntu

pico2 开发环境搭建-基于ubuntu 安装编译工具链下载sdk 和example编译example 安装编译工具链 sudo apt install cmake gcc-arm-none-eabi libnewlib-arm-none-eabi libstdc++-arm-none-eabi-newlib 注意cmake的版本,需要在3.17 以上 下载sdk 和ex

pip-tools:打造可重复、可控的 Python 开发环境,解决依赖关系,让代码更稳定

在 Python 开发中,管理依赖关系是一项繁琐且容易出错的任务。手动更新依赖版本、处理冲突、确保一致性等等,都可能让开发者感到头疼。而 pip-tools 为开发者提供了一套稳定可靠的解决方案。 什么是 pip-tools? pip-tools 是一组命令行工具,旨在简化 Python 依赖关系的管理,确保项目环境的稳定性和可重复性。它主要包含两个核心工具:pip-compile 和 pip

跨系统环境下LabVIEW程序稳定运行

在LabVIEW开发中,不同电脑的配置和操作系统(如Win11与Win7)可能对程序的稳定运行产生影响。为了确保程序在不同平台上都能正常且稳定运行,需要从兼容性、驱动、以及性能优化等多个方面入手。本文将详细介绍如何在不同系统环境下,使LabVIEW开发的程序保持稳定运行的有效策略。 LabVIEW版本兼容性 LabVIEW各版本对不同操作系统的支持存在差异。因此,在开发程序时,尽量使用

Go Playground 在线编程环境

For all examples in this and the next chapter, we will use Go Playground. Go Playground represents a web service that can run programs written in Go. It can be opened in a web browser using the follow