如何在CUDA中限制GPU内存大小

How to restrict the GPU memory size in CUDA?

本文关键字:GPU 内存 CUDA      更新时间:2023-10-16

我正在实现一个核心外的算法。我可用的GPU内存大小是4GB。然而,为了在较小的数据集上测试和比较我的算法与其他版本,我想限制我可用的GPU上的内存大小。有什么办法可以做到吗?

(我总是不能使用大量的内存存在,然而,我必须表明,如果其他版本使用的内存超过可用的内存,那么其他版本崩溃,而我的算法继续运行的核心,因此它将是伟大的,如果有一种方法来限制GPU上可用的全局内存大小,编程或其他方式)。

一种解决方案是cudaMalloc一个虚拟指针,其大小与您希望在程序开始时从可用内存中取出的大小相同。