文件名称:rmm:RAPIDS内存管理器
文件大小:430KB
文件格式:ZIP
更新时间:2024-04-05 23:18:21
cuda memory-management memory-allocation rapids C++
RMM:RAPIDS内存管理器 在以GPU为中心的工作流程中实现最佳性能通常需要自定义主机和设备内存的分配方式。 例如,使用“固定”主机内存进行异步主机<->设备内存传输,或使用设备内存池子分配器来减少动态设备内存分配的成本。 RAPIDS内存管理器(RMM)的目标是提供: 通用接口,可自定义和内存分配 接口的的集合 使用该接口进行内存分配的的集合 有关RMM提供的接口以及如何在C ++代码中使用RMM的信息,请参见。 注意:有关最新的稳定确保您位于main分支上。 安装 conda RMM可以从rapidsai频道与Conda( 或的完整)一起安装: # for CUDA 10.2 conda install -c nvidia -c rapidsai -c conda-forge -c defaults \ rmm cudatoolkit=10.2 # for CUD