matlab - MatLab 的 Cuda API

Question

我最近参与了实现基于 GPU 的图像配准技术。cpu 代码是用 MatLab 编写的，这就是我可以从密集搜索中得到的，所以我坚持使用 MatLab。我有夹克或 matlab ptx 文件的两种选择。

最近 mathwork 收购了夹克，所以我唯一的选择是将我的内核编写为 CU 文件并用于nvcc -ptx filename.cu生成ptx 文件。我担心的是，我错过了很好的机会，因为我没有使用 Cuda API 或引入的库来解决 GPU 上的一些基本矩阵运算，比如 CUBLAS。

那么有没有办法使用 MatLab 中的 Cuda API，mex 文件呢？是否有一个用 C/C++ 编写的好的图像注册框架，我可以在我的研究中使用，而不是我正在使用的 MatLab 版本？

Answer 1

这篇文章提供了一个关于如何编译 CUDA c/c++ mex 代码的教程。

您的问题类似于 在 linux 中编写 CUDA+MEX 代码的一种处理方式吗？ 那里的答案可能会对您有所帮助。

有关 CUDA API 和 CUDA 库的参考手册，请参阅CUDA 工具包文档。

Answer 2

虽然工作量更大，但您可以通过创建MEX 函数直接使用 CUDA 源代码。

OpenCV是一个很好的通用计算机视觉库，用 C++ 编写。根据您正在进行的注册类型，它可能会有所帮助。