我在这里对 Numba 做错了什么？

Question

我正在尝试学习如何使用 Numba 模块。到目前为止，由于与 NumPy 接口的一些问题，我还没有得到任何工作。这是我正在运行的代码（来自 Numba 文档）和我得到的错误：

from numba import jit
import numpy as np

x = np.arange(100).reshape(10, 10)

@jit(nopython=True) # Set "nopython" mode for best performance, equivalent to @njit
def go_fast(a): # Function is compiled to machine code when called the first time
    trace = 0.0
    for i in range(a.shape[0]):   # Numba likes loops
        trace += np.tanh(a[i, i]) # Numba likes NumPy functions
    return a + trace              # Numba likes NumPy broadcasting

print(go_fast(x))

    Traceback (most recent call last):
File "C:/Users/JoHn/Documents/Current Classes/MEEN575_Optimization/HW6/Optimal_controller/angle_wrapping.py", line 84, in <module>
print(go_fast(x))
TypeError: expected dtype object, got 'numpy.dtype[float64]'

我从一些搜索中知道，这曾经是或现在是最近的一个已知错误，并且与 Numba 的新版本需要更新的 NumPy 或类似的东西有关，但据我所知，我有最新的NumPy 构建，版本 1.20。关于我做错了什么的任何提示？需要明确的是，我从来没有很好地理解如何在 python 中干净地设置环境，所以很可能我只是在这里遗漏了一些明显的东西。

Answer 1

更新到 0.53.1 有效。对我来说，它在 0.47.x 上也失败了。似乎更多的是 numpy 问题。解决安装 numpy >=1.20.0 和 numba v>0.52 的一种方法。

有关此问题的更多信息： https://github.com/numba/numba/issues/6041

P.S：不知道你是否还有这个错误，只是想更新，遇到了类似的问题。