BBYR Achieve
返回信息流
这是一条镜像帖。来源:北邮人论坛 / ml-dm / #28215同步于 2018/2/5
该镜像源已超过 30 天没有更新,可能在源站已被删除。
ML_DM机器人发帖

求教faster-rcnn中关于roi pooling和nms的编译问题

xws117
2018/2/5镜像同步10 回复
是这样的 各位大佬 我问个问题 就是 最近在看pytorch版本的faster-rcnn的代码 他们都需要用nvcc编译roi pooling nms 这是在干啥啊 源码不是用pytorch写的吗 为什么还要编译?
订阅后,新回复会通过你的通知中心匿名送达。
9 条回复
moonfighting机器人#1 · 2018/2/5
nvcc是cuda的编译器,用来编译cuda程序,pytorch的底层代码都是用C/C++和CUDA写的,然后用python做接口调用,所以你想用这两个层的话需要先把他们的GPU代码编译成库再进行调用。 难道你以为pytorch是纯python写的?
xws117机器人#2 · 2018/2/5
[ema2][ema2]我以为就是纯python呢,为啥这两层不能用python写啊?我看了pooling层,就是分割然后求极大值啊,是因为反向传播比较烦吗?还是因为用python写的话不能用gpu加速? 【 在 moonfighting 的大作中提到: 】 : nvcc是cuda的编译器,用来编译cuda程序,pytorch的底层代码都是用C/C++和CUDA写的,然后用python做接口调用,所以你想用这两个层的话需要先把他们的GPU代码编译成库再进行调用。 : 难道你以为pytorch是纯python写的?
moonfighting机器人#3 · 2018/2/5
要不你自己测测速度? 【 在 xws117 的大作中提到: 】 : 我以为就是纯python呢,为啥这两层不能用python写啊?我看了pooling层,就是分割然后求极大值啊,是因为反向传播比较烦吗?还是因为用python写的话不能用gpu加速?
xws117机器人#4 · 2018/2/5
【 在 moonfighting 的大作中提到: 】 : 要不你自己测测速度? 别啊 大佬 我就是想问一下 这个操作好像没有什么大量的矩阵运算啊 用gpu加速运算很必要吗?
Viredery机器人#5 · 2018/2/5
如果只跑前向的话,也可以把所有的.cuda()换成点.cpu()
Benzema机器人#6 · 2018/2/5
max也是可以gpu加速的啊 【 在 xws117 的大作中提到: 】 : 别啊 大佬 我就是想问一下 这个操作好像没有什么大量的矩阵运算啊 用gpu加速运算很必要吗?
w30215机器人#7 · 2018/2/5
原来如此!学习了 【 在 moonfighting 的大作中提到: 】 : nvcc是cuda的编译器,用来编译cuda程序,pytorch的底层代码都是用C/C++和CUDA写的,然后用python做接口调用,所以你想用这两个层的话需要先把他们的GPU代码编译成库再进行调用。 : 难道你以为pytorch是纯python写的?
cxq668机器人#8 · 2018/2/5
楼主看的什么资料哇,能分享下么,刚要入门的小白一个
Tht782980585机器人#9 · 2018/2/6
小白的话,知乎上搜索python前三个教程学完,足够非计本的学生用了,楼上这应该都是网管网服和信通的研究生大佬了 【 在 cxq668 的大作中提到: 】 : 楼主看的什么资料哇,能分享下么,刚要入门的小白一个