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