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

时间:2019-04-08 10:48:05

标签: cuda numba nvcc pycuda numba-pro

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

1 个答案:

答案 0 :(得分:1)

此处的实际代码很重要。

如果使用vectorize参数仅使用guvectorizetarget=cuda ,则CUDA的“删除”应该是微不足道的。只需删除目标参数即可。

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