本文介绍了如何使用CUDA运行时API中的cudaMemGetInfo函数获取当前设备上的空闲内存和总内存。此外，还提到了NVIDIA Management Library (NVML) 中的nvmlDeviceGetMemoryInfo函数，该函数可以获取设备上已用、空闲和总内存的字节数。对于AMD设备，文中未提及等效函数。
cudaMemGetInfo (documented here) requires nothing other than the cuda runtime API to get free memory and total memory on the current device.
Check out the function nvmlDeviceGetMemoryInfo in NVIDIA Management Libraryhttps://developer.nvidia.com/nvidia-management-library-nvml:
"Retrieves the amount of used, free and total memory available on the device, in bytes."
Don't know if AMD has something equivalent.
D3DKMTQueryStatistics is what you need.
扫一扫
249
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
