相关疑难解决方法(0)

我想知道如何使用 Python、Numba 和 cudatoolkit 获取 GPU 中的 CUDA 核心总数。

python中是否有安装了numba或tensorflow的代码？例如，如果我想知道 GPU 内存信息，我可以简单地使用：

from numba import cuda
gpus = cuda.gpus.lst
for gpu in gpus:
    with gpu:
        meminfo = cuda.current_context().get_memory_info()
        print("%s, free: %s bytes, total, %s bytes" % (gpu, meminfo[0], meminfo[1]))

在 numba。但是我找不到任何代码可以为我提供每个块信息的最大线程数。我希望代码能够检测每个块的最大线程数，并进一步计算每个方向上指定的块数。