Pycharm 在一个简单的示例中，Numba矢量化无法正常工作

我正在尝试为我正在工作的一个程序添加gpu支持，我遇到了numba和cuda，然后是ufunc。我将从中学习基本示例 我正在使用pycharm 2019.3.1 我用的是麻木0.51.2

我得到一个错误：

Traceback (most recent call last): line 19, in <module>
D = cu_discriminant(A, B, C)
TypeError: not enough arguments for format string

它在target='cpu'时工作，但在target='cuda'时不工作

---

我不知道该怎么做，因为GPU加速会大大加快我的代码速度。非常感谢您的帮助。

您的代码在没有“tensorflow 2”的情况下在两种情况下都能正常工作

结果如下

如果目标为“cpu” [10.905523 10.944542 10.116297 ... 10.752323 10.669133 10.494761] 花了0.083秒

如果目标为“cuda” [10.605754 10.977879 10.845842 ... 10.56295 10.10678 10.774803] 花了0.129秒

# with rtx 3070

但是我在安装tensorflow 2的地方发现了同样的错误。

---

错误消息是NvvmError:编译失败

您使用的是哪个GPU、CUDA和Python版本？也许更一般地说，你的错误不一定来自麻木。这是一个通用的Python错误。我之所以提到这两个，是因为你的代码在我的机器上运行得很好——Python 3.7、CUDA 10.2、V100 GPU、Numba 0.51.2.2080，我不知道如何正确安装CUDA——我用c编写了vs代码。Python3.7.在终端输入的nvidia smi向您展示了什么？这似乎不是答案。请在此处查看如何写出精彩答案：

# with rtx 3070