cuda - 如何从 python 项目中禁用或删除 numba 和 cuda？

Question

我已经克隆了一个“PointPillars”repo，仅使用点云作为输入进行 3D 检测。但是当我开始运行它时，我注意到它使用 cuda 和 numba。有了关于这两者的任何先验知识，我问是否有任何方法可以删除或禁用 numba 和 cuda。我想在本地服务器上仅使用 CPU 运行它，所以我希望您的建议能够解决。

Answer 1

实际代码在这里很重要。

如果仅使用vectorize或guvectorize使用target=cuda参数，那么“删除”CUDA 应该是微不足道的。只需删除目标参数。

但是，如果使用@cuda.jit装饰器，或在主机和设备之间显式复制数据，则将涉及其他代码重构。在这种情况下，这里没有简单的答案，必须通过重构或移植将代码转换为替代的串行或并行实现。