|
目的:将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. 要实现我的目的,是否有比上两个方法更简单的方案。
|
|