如何验证@jit()装饰器在GPU上正常工作?

时间:2018-12-09 20:57:37

标签: python gpu numba

在处理我的代码时,即使使用nopython = True选项,使用@jit装饰器后也找不到任何加速。

我怀疑jit确实支持GPU,因为我在其参数中找不到此选项。我还认为parallel = True选项适用于CPU内核。

有任何点击,更正吗? 如何验证我的代码可在GPU上正常工作?

非常感谢:)

0 个答案:

没有答案