|
|
目的：将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. 要实现我的目的，是否有比上两个方法更简单的方案。
|
|
收藏
闽公网安备 35010202001061号