内存分配 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）。