我只使用点云作为输入克隆了一个“PointPillars”回购用于3D检测。但是当我来运行时,我注意到它使用了cuda和numba。有了这两个先前的知识,我问是否有任何方法可以删除或禁用numba和cuda。我想在只有CPU的本地服务器上运行它,所以我希望你的建议能够解决。
实际的代码在这里很重要。
如果使用vectorize
参数仅使用guvectorize
或target=cuda
,那么“删除”CUDA应该是微不足道的。只需删除目标参数即可。
但是,如果使用@cuda.jit
装饰器,或者在主机和设备之间显式复制数据,那么将涉及其他代码重构。在这种情况下,没有简单的答案,代码必须通过重构或移植转换为备用串行或并行实现。