Toybrick

标题: rknn_roolkit_v1.3.0+pytorch1.2.0配置问题 [打印本页]

作者: rlgl 时间: 2020-3-30 10:45
标题: rknn_roolkit_v1.3.0+pytorch1.2.0配置问题
目的：将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 时间: 2020-3-31 10:51
跟新尝试：
pc端安装ubuntu16.04系统，按照Rockchip_User_Guide_RKNN_Toolkit_V1.3.0_CN.pdf文档中的3.1.2节：“通过dock进行安装”，运行镜像的结果如下

作者: jefferyzhang 时间: 2020-3-31 14:29

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 时间: 2020-3-31 14:56

jefferyzhang 发表于 2020-3-31 14:29
docker里没有pytorch可以自己pip install

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

作者: jefferyzhang 时间: 2020-3-31 15:04

rlgl 发表于 2020-3-31 14:56
那请问import tensorflow和from rknn.api import RKNN会报illegal instruction(core dumped)错误，这个怎 ...

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

欢迎光临 Toybrick (https://t.rock-chips.com/)