如果我们没有明确提到目标，@jit是否在gpu上运行python代码？

我有一个python代码，我正在尝试使用Cuda加速。我使用了@jit函数。如何知道代码是否在gpu上实际运行？有什么方法可以检查/验证吗？

您应该使用numba.cuda.jit在GPU上运行jitted函数。此外，这个函数应该以CUDA内核（）的方式编写。当函数运行时，可以使用一些GPU监视器（例如nvidia smi for Linux）来查看GPU负载并检查其是否参与计算