最近需要用 mxnet 这个框架,所以在 ubuntu 16.04 上装了一个,前面采过的坑网上基本都有提到,就不说了。nnpack 是用来做CPU加速的,在按照 mxnet 官方文档配置的时候,遇到一些坑,网上没有提到,所以记录一下。
我是按照文档 http://mxnet.io/how_to/nnpack.html 里面的配置过程来进行配置的,到最后一步 ninja 编译的时候报了很多错,大概是这样的:
TypeError: 'float' object cannot be interpreted as an integer
TypeError: Unsupported operand: 24.0
错误打印的信息里面可以看到出错的源文件和位置,通过追查代码可以发现这样的错误都是由于没有把计算得到的浮点类型转换成整型,所以在 nnpack 的代码里面对应的地方加上int()转换一下就可以了。
我猜测这是版本的问题,但是不能确定是 peachpy 还是 nnpack 的问题,以后有空再追查一下吧。