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

How to disable or remove numba and cuda from python project?

我克隆了一个 "PointPillars" 存储库,用于仅使用点云作为输入的 3D 检测。但是当我来到 运行 它时,我注意到它使用了 cuda 和 numba。有了关于这两者的任何先验知识,我想问一下是否有任何方法可以删除或禁用 numba 和 cuda。我只想 运行 在本地服务器上 CPU ,所以我希望你的建议能够解决。

实际代码在这里很重要。

如果用法是only of vectorize or guvectorize using the target=cuda parameter, then "removal" of CUDA should是微不足道的。只需删除目标参数。

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