numba nopython mode 不支持numpy.int类型

最新推荐文章于 2024-04-07 15:44:47 发布

原创

最新推荐文章于 2024-04-07 15:44:47 发布 · 4.6k 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#python #numba

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

一则小记，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.
    array