背景
- 使用Anaconda配置源码环境
- 源码需要使用
python setup.py
来编译依赖cuda的torch拓展模块,如 nms,ROIPool,ROIAlign等等
- 系统的CUDA和Conda装的
cudatoolkit
版本不同
问题
符合上述背景条件或者类似条件,会导致一些奇怪的错误,例如:
ImportError: ***/ATSS/atss_core/_C.cpython-36m-x86_64-linux-gnu.so: undefined symbol: __cudaRegisterFatBinaryEnd
<