在使用CenterNet过程中
1.先是在DCNv2编译的时候报错:
In file included from /home/leo/anaconda3/envs/cn/lib/python3.7/site-packages/torch/include/ATen/cuda/CUDAContext.h:5:0,
from /home/leo/Desktop/wjl/detect/pytorch_simple_CenterNet_45-master/lib/DCNv2/src/cpu/dcn_v2_cpu.cpp:4:
/home/leo/anaconda3/envs/cn/lib/python3.7/site-packages/torch/include/c10/cuda/CUDAStream.h:6:30: fatal error: cuda_runtime_api.h: No such file or directory
compilation terminated.
由于是未找到cuda_runtime_api.h文件,因此我直接更改了setup.py:
然后编译通过但是测试中出现问题2.
2.在运行test.py 测试的时候报错:
ImportError: dynamic module does not define module export function (PyInit__ext)
我已经检查cuda10.0的环境变量没有问题,只是cuda runtime API版本是 cuda10.0,driver API版本是cuda10.1,还不清楚是不是这个的原因导致的.
有没有懂的兄弟伸个援手.