在使用GPU版本的Paddle3D时候,想用caddn算法推理一下,然后infer.py 之后出现No module named 'iou3d_nms'
报错是:No module named 'iou3d_nms',还有这个:command 'C:\Program Files (x86)\Microsoft Visual Studio\2022\BuildTools\VC\Tools\MSVC\14.37.32822\bin\HostX86\x64\cl.exe' failed with exit code 2 2023-09-06 21:49:52,429 - ERROR - iou3d_nms builded fail!,感觉是没有装nms_gpu这个模块,没办法import,找了半天也没找到这个py
paddle验证安装:
import paddle C:\Users\Yuanqing288.conda\envs\paddle\lib\site-packages\setuptools\sandbox.py:13: DeprecationWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html import pkg_resources C:\Users\Yuanqing288.conda\envs\paddle\lib\site-packages\pkg_resources_init_.py:2871: DeprecationWarning: Deprecated call to
pkg_resources.declare_namespace('google'). Implementing implicit namespace packages (as specified in PEP 420) is preferred topkg_resources.declare_namespace. See https://setuptools.pypa.io/en/latest/references/keywords.html#keyword-namespace-packages declare_namespace(pkg) C:\Users\Yuanqing288.conda\envs\paddle\lib\site-packages\pkg_resources_init_.py:2871: DeprecationWarning: Deprecated call topkg_resources.declare_namespace('mpl_toolkits'). Implementing implicit namespace packages (as specified in PEP 420) is preferred topkg_resources.declare_namespace. See https://setuptools.pypa.io/en/latest/references/keywords.html#keyword-namespace-packages declare_namespace(pkg) paddle.utils.run_check() Running verify PaddlePaddle program ... W0906 21:55:17.421447 54172 gpu_resources.cc:61] Please NOTE: device: 0, GPU Compute Capability: 8.6, Driver API Version: 12.2, Runtime API Version: 11.7 W0906 21:55:17.434427 54172 gpu_resources.cc:91] device: 0, cuDNN Version: 8.4. PaddlePaddle works well on 1 GPU. PaddlePaddle works well on 1 GPUs. PaddlePaddle is installed successfully! Let's start deep learning with PaddlePaddle now. paddle.device.is_compiled_with_cuda() True
需要先编译一下自定义算子iou3d_nms,在paddle3d/ops/目录下执行 python setup.py install 可以编译paddle3d的所有自定义算子。