ImportError: /home/zdk/CenterFusion-master/src/lib/model/networks/DCNv2/_ext.cpython-37m-x86_64-linux-gnu.so: undefined symbol: _Z20dcn_v2_cuda_backwardRKN2at6TensorES2_S2_S2_S2_S2_iiiiiiiii
时间: 2023-08-10 07:07:10 浏览: 185
ImportError: DLL load failed while importing cv2: 找不到指定的模块(亲自解决)
这个错误通常是由于 PyTorch 和 DCNv2 的 CUDA 版本不兼容导致的。你可以尝试以下几种解决方案:
1. 确认 PyTorch 和 DCNv2 的 CUDA 版本是否匹配,如果不匹配则需要重新编译 DCNv2。
2. 确认 DCNv2 是否正确地安装,可以尝试重新安装或者升级 DCNv2。
3. 确认是否正确设置了环境变量,包括 LD_LIBRARY_PATH 和 PATH 等。
4. 如果以上方案都不行,可以尝试降低 PyTorch 和 DCNv2 的版本或者使用其他的实现方式。
阅读全文