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

问题描述 投票:-2回答:1

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

cuda numba nvcc pycuda numba-pro
1个回答
1
投票

实际的代码在这里很重要。

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

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

© www.soinside.com 2019 - 2024. All rights reserved.