numba nopython mode 不支持numpy.int类型

最新推荐文章于 2023-12-14 15:50:24 发布

风海流

最新推荐文章于 2023-12-14 15:50:24 发布

阅读量4.4k

点赞数

分类专栏： python 文章标签： python numba

本文链接：https://blog.csdn.net/huyaoyu/article/details/106465104

版权

本文记录了在使用Numba库时遇到的问题，即Numba的nopython模式不支持NumPy的int类型。解决方案是明确指定整型的位宽，如使用`numpy.int32`或`numpy.int64`。测试环境为Ubuntu 18.04，Python 3.6.8，Numba 0.46.0，NumPy 1.16.4。

摘要由CSDN通过智能技术生成

一则小记，numba库并不支持NumPy的int类型，需要指定整型的位宽，使用int32和int64都没有问题。测试代码如下

本机系统Ubuntu 18.04, Python 3.6.8, numba 0.46.0, NumPy 1.16.4

import numba
import numpy as np
import sys

@numba.jit(nopython=True)
def test_zeros(H, W):
    # array = np.zeros((H, W, 3), dtype=np.int) # Will cause jit error