CUDA:有人试过把gpumat传递给其他Python CUDA模块吗?

2024-06-06 08:43:00 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在尝试加速一些需要当前任何python CUDA模块中都不可用的任务的图像处理。例如,^{}有许多numpy/scipy函数在OpenCV中不可用。我很好奇是否有人成功地在不同的代码基之间传递了GPU上的数据,而不是花费大量的计算时间在CPU和GPU之间来回传递来进行转换?我对cupy特别感兴趣,但可以从其他代码的任何经验开始。你知道吗


Tags: 模块数据函数代码numpygpu时间scipy