c - 如何在 linux KNN CUDA 上编译？

Question

最近，我发现knn CUDA这是一组基于暴力实现 knn 搜索的 Mex 文件，但在 README.md 中我没有找到使用 linux 发行版在 matlab 中编译这些文件的方法。我将不胜感激关于如何处理这个问题的想法。

Answer 1

我是这个 kNN 代码的作者:)

早在 2008 年，代码是使用 Windows XP 操作系统编写的。 由于我提供了源代码，您应该能够生成 linux mex 文件。 在自述文件中，我为 Windows 提供以下命令行:

nvmex -f nvmexopts.bat knn_cuda_with_indexes.cu -I'C:\CUDA\include' -L'C:\CUDA\lib' -lcufft -lcudart -lcuda -D_CRT_SECURE_NO_DEPRECATE

根据您的 Linux 发行版对其进行调整以生成您的 mex 文件。 5 年内可能发生了很多变化，因此您可能需要修改一些内容。 然而，我从用户那里得到的反馈表明它工作得很好。

也尝试阅读有关如何在 Linux 下编译 CUDA 代码的信息。 我想 NVidia 提供了一个非常好的教程。