Toybrick

标题: Pytorch现在支持吗 [打印本页]

作者: 运河边边上 时间: 2020-3-14 18:32
标题: Pytorch现在支持吗

pytorch用的比较多，装了半天还是装不上。。。之前看帖子说还没支持。。

作者: jefferyzhang 时间: 2020-3-15 10:24
rknn 1.3 支持 pytorch1.2 版本

作者: 运河边边上 时间: 2020-3-16 09:39

jefferyzhang 发表于 2020-3-15 10:24
rknn 1.3 支持 pytorch1.2 版本

大佬我的意思是：板子上python能装pytorch？ import torch 使用吗

作者: jefferyzhang 时间: 2020-3-16 09:41

运河边边上发表于 2020-3-16 09:39
大佬我的意思是：板子上python能装pytorch？ import torch 使用吗

当然可以，我们板子也是通用架构+发行版Linux，能装都能跑。
只是跑的CPU而已

作者: 运河边边上 时间: 2020-3-16 09:43

jefferyzhang 发表于 2020-3-16 09:41
当然可以，我们板子也是通用架构+发行版Linux，能装都能跑。
只是跑的CPU而已 ...

好得

作者: zengwubin 时间: 2020-3-21 18:34
请问你是怎么安装torch的，我安装了一个运行是报错ImportError: libcudart.so.10.0: cannot open shared object file: No such file or directory，知道怎么解决吗？

作者: 运河边边上 时间: 2020-3-26 14:40

zengwubin 发表于 2020-3-21 18:34
请问你是怎么安装torch的，我安装了一个运行是报错ImportError: libcudart.so.10.0: cannot open shared o ...

我还没装成功呢

作者: rlgl 时间: 2020-3-31 11:04

jefferyzhang 发表于 2020-3-16 09:41
当然可以，我们板子也是通用架构+发行版Linux，能装都能跑。
只是跑的CPU而已 ...

请问板子上（rk3399proD）是怎么装pytorch的，pytorch官方好像没发布arm端的安装包，我使用的命令是
pip install torch==1.2.0+cpu torchvision==0.4.0+cpu -f https://download.pytorch.org/whl/torch_stable.html --user
提示错误：No matching distribution found for torch==1.2.0+cpu。
我现在板子上rknn_toolkit_1.3.0已经配置好了，./example/tflite中的示例也可以成功运行，但./example/pytorch中的示例无法运行，pytorch和torchvision都安装不上，求大佬解答下，被这问题折磨一周了已经，尝试了很多方法，试过的方法写在这个帖子里，目前还没人回应http://t.rock-chips.com/forum.ph ... tid=1420&extra=

作者: jefferyzhang 时间: 2020-3-31 11:16

rlgl 发表于 2020-3-31 11:04
请问板子上（rk3399proD）是怎么装pytorch的，pytorch官方好像没发布arm端的安装包，我使用的命令是
pip ...

我们是芯片公司。。。不会去干预debian10的包管理和社区维护内容。。
这种东西你们可以去google pytorch1.2有没有arm版本的来安装，没有的话就在pc上搞就可以了。
不过既然你们提出来了，我们回头有空可以去收集下arm版pytorch

作者: rlgl 时间: 2020-3-31 13:44

jefferyzhang 发表于 2020-3-31 11:16
我们是芯片公司。。。不会去干预debian10的包管理和社区维护内容。。
这种东西你们可以去google pytorch1 ...

谢谢您的回复，我再问个问题

我的目的是：将pytorch模型转为rknn模型，然后在rk3399prod开发板上进行推理
您提到可以在pc上搞，我理解的意思是不是这样做
1. pc上配置rknn_toolkit_v1.3.0+pytorch1.2.0，然后在pc上将pytorch模型转为rknn模型，此时仅仅需要pc就可，不需要将开发板连接到pc;
2. 开发板配置好rknn_toolkit_v1.3.0环境，将转换后的rknn模型拷贝到开发版上进行推理。

目前我正在pc端尝试配置rknn_toolkit_v1.3.0+pytorch1.2.0，win10和ubuntu16.04上都在尝试，参考“Rockchip_User_Guide_RKNN_Toolkit_V1.3.0_CN.pdf”、“Rockchip_Quick_Start_RKNN_Toolkit_V1.3.0_CN.pdf”这两个文档去安装，最让我不解的是按照Rockchip_User_Guide_RKNN_Toolkit_V1.3.0_CN.pdf文档中的3.1.2节：“通过dock进行安装”，也无法成功，提示illegal instruction(core dumped)失败截图我上传到这个帖子了http://t.rock-chips.com/forum.ph ... tid=1420&extra=

作者: jefferyzhang 时间: 2020-3-31 14:28
是的，没问题，而且我们也推荐在PC上转换rknn。

作者: Leonod 时间: 2020-12-4 11:45

jefferyzhang 发表于 2020-3-15 10:24
rknn 1.3 支持 pytorch1.2 版本

rknn 1.3 支持 pytorch1.2 版本
->是指只支持转换pytorch1.2训练出来的模型吗？如果有pt文件的话，我看rknntool提供的接口在转换和推理时是用不到pytorch的，希望得到解答。

作者: jefferyzhang 时间: 2020-12-4 14:55

Leonod 发表于 2020-12-4 11:45
rknn 1.3 支持 pytorch1.2 版本
->是指只支持转换pytorch1.2训练出来的模型吗？如果有pt文件的话，我看 ...

当然是模型，我们怎么会去跑pytorch

作者: Leonod 时间: 2020-12-4 17:48

jefferyzhang 发表于 2020-12-4 14:55
当然是模型，我们怎么会去跑pytorch

好的，感谢回答!

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