wuzhihao7788 / yolodet-pytorch

reproduce the YOLO series of papers in pytorch, including YOLOv4, PP-YOLO, YOLOv5,YOLOv3, etc.
390 stars 106 forks source link

PPYOLO: ImportError: cannot import name 'deform_conv_cuda' from 'yolodet.ops.dcn' #5

Open lyss111 opened 3 years ago

lyss111 commented 3 years ago

新建了一个虚拟环境,用python setup.py develop编译DCN时 :RuntimeError: The NVIDIA driver on your system is too old (found version 10020). 我这版本也不老啊😢 用pip install -v -e .编译DCN时 InstallationError: Command errored out with exit status 1: python setup.py egg_info Check the logs for full command output.

wuzhihao7788 commented 3 years ago

No description provided.

这个问题应该是DCN的pytroch cuda包没有编译成功,建议你将环境卸载,build文件包和yolodet/ops/dcn下的.os文件删除重新编译。

wuzhihao7788 commented 3 years ago

新建了一个虚拟环境,用python setup.py develop编译DCN时 :RuntimeError: The NVIDIA driver on your system is too old (found version 10020). 我这版本也不老啊😢 用pip install -v -e .编译DCN时 InstallationError: Command errored out with exit status 1: python setup.py egg_info Check the logs for full command output.

我的环境是显卡2080Ti cuda 10.2,编译没有问题。编译过程我也不是一次性成功的,也是反复三次成功的。需要慢慢尝试,不要慌张,做深度学习,解决环境问题,也是我们重要的技能之一哈