标签: python python-3.x cuda gpu numba
我刚开始使用Numba并使用了Mandelbrot示例,然后查看它是否在GPU上运行,我打开了任务管理器,发现代码使用的是集成GPU(GPU 0),而不是我喜欢的GPU (GPU 1)。如何强制Numba使用GPU 1?
代码使用@jit装饰器。
@jit