Toybrick

rknn_roolkit_v1.3.0+pytorch1.2.0配置问题

rlgl

注册会员

积分
101
楼主
发表于 2020-3-30 10:45:36    查看: 6063|回复: 4 | [复制链接]    打印 | 只看该作者
目的:将pytorch模型使用rknn_toolkit转为rknn模型需要做:要使用rknn包中的load_pytorch方法,则需要rknn_Toolkit_v1.3.0+pytorch1.2.0环境


尝试过的方法:
1. rk3399Pro开发板上配置debian10+rknn_Toolkit_v1.3.0+pytorch1.2.0;
2. pc端配置win10+rknn_roolkit_v1.3.0+pytorch1.2.0。

结果:
1. 开发板上成功配置debian10+rknn_Toolkit_v1.3.0,但无法配置pytorch1.2.0,
使用的命令为:pip install torch==1.2.0+cpu torchvision==0.4.0+cpu -f https://download.pytorch.org/whl/torch_stable.html --user
错误信息为:“no matching dirtribution found for torch==1.2.0+cu92”
网上查的原因:pytorch官方发布的安装包似乎只支持x86/amd架构的安装包,未发布arm端的安装包
2. win10上使用conda按照"Rockchip_Quick_Start_RKNN_Toolkit_V1.3.0_CN.pdf"文档中的第四节“windows平台快速上手指南”
配置rknn_roolkit_v1.3.0+pytorch1.2.0,目前还未成功

我的问题:
1. 目前开发板上是否能安装pytorch,若可以,求安装方法;
2. win10上如何配置rknn_roolkit_v1.3.0+pytorch1.2.0环境。若环境配置好了,可否在pc端将pytorch模型转为rknn模型(不连接开发板的情况下),
再将转换后的rknn模型拷贝到开发板上进行下一步工作;

3. 要实现我的目的,是否有比上两个方法更简单的方案。
回复

使用道具 举报

rlgl

注册会员

积分
101
沙发
 楼主| 发表于 2020-3-31 10:51:09 | 只看该作者
跟新尝试:
pc端安装ubuntu16.04系统,按照Rockchip_User_Guide_RKNN_Toolkit_V1.3.0_CN.pdf文档中的3.1.2节:“通过dock进行安装”,运行镜像的结果如下

本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有帐号?立即注册

x
回复

使用道具 举报

jefferyzhang

版主

积分
12925
板凳
发表于 2020-3-31 14:29:16 | 只看该作者
rlgl 发表于 2020-3-31 10:51
跟新尝试:
pc端安装ubuntu16.04系统,按照Rockchip_User_Guide_RKNN_Toolkit_V1.3.0_CN.pdf文档中的3.1.2 ...

docker里没有pytorch可以自己pip install
回复

使用道具 举报

rlgl

注册会员

积分
101
地板
 楼主| 发表于 2020-3-31 14:56:30 | 只看该作者
jefferyzhang 发表于 2020-3-31 14:29
docker里没有pytorch可以自己pip install

那请问import tensorflow和from rknn.api import RKNN会报illegal instruction(core dumped)错误,这个怎么解决(图中也运行了这两条命令)
回复

使用道具 举报

jefferyzhang

版主

积分
12925
5#
发表于 2020-3-31 15:04:41 | 只看该作者
rlgl 发表于 2020-3-31 14:56
那请问import tensorflow和from rknn.api import RKNN会报illegal instruction(core dumped)错误,这个怎 ...

我们这里测试没有遇到过,你至少也得稍微gdb之类的跟踪下堆栈再报吧,就发一个illegal instruction咋知道啥问题。
最好自己能装环境,就不要用docker了。。
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


快速回复 返回顶部 返回列表