opencl - 内存分配 Nvidia vs AMD

Question

我知道 AMD GPU 上的单个 GPU 内存块有 128MB 的限制。Nvidia GPU 是否有类似的限制？

Answer 1

您可以在运行时使用 clGetDeviceInfo 和 CL_DEVICE_MAX_MEM_ALLOC_SIZE 查询此信息。

有关详细信息，请参阅clGetDeviceInfo 手册页。

Answer 2

在 GTX 560clGetDeviceInfo上为 CL_DEVICE_MAX_MEM_ALLOC_SIZE 返回 256MiB，但是我可以分配略小于 1GiB。请参阅讨论该问题的线程。

然而，在 AMD 上，这个限制是强制的。您可以通过更改 GPU_MAX_HEAP_SIZE 和 GPU_MAX_ALLOC_SIZE 环境变量来提高它（请参阅此线程）。