内存分配Nvidia vs AMD

Question

我知道AMD GPU上的一块GPU内存有128MB的限制。 Nvidia GPU有没有类似的限制？

Answer 1

在GTX 560 clGetDeviceInfo为CL_DEVICE_MAX_MEM_ALLOC_SIZE返回256MiB，但是我可以分配稍微小于1GiB。讨论这个问题见this thread。

在AMD上，但是这个限制被强制执行。您可以通过更改GPU_MAX_HEAP_SIZE和GPU_MAX_ALLOC_SIZE环境变量来提高它（请参阅this thread）。

Answer 2

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

请参阅clGetDeviceInfo Man Page了解更多信息。