内存分配Nvidia vs AMD

Question

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

Answer 1

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

有关详细信息，请参阅clGetDeviceInfo Man Page。

Answer 2

在GTX 560上clGetDeviceInfo为CL_DEVICE_MAX_MEM_ALLOC_SIZE返回256MiB，但是我可以分配略低于1GiB的值。请参阅this thread讨论此问题。

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