Thundernet_Pytorch icon indicating copy to clipboard operation
Thundernet_Pytorch copied to clipboard
verified publisher icon DayBreak-u

PSROI Align 模块

Open harryQAQ opened this issue 6 years ago • 4 comments

你好，非常感谢你的工作，我在build psroialign模块之后跑不通，提示cudaCheckError() failed : invalid device function，在setup.py中新增了"nvcc":['-arch=sm_61']，但还是没有解决，请问你有遇到这样的情况吗？

harryQAQ avatar Nov 29 '19 13:11 harryQAQ

你好，你用的torch是什么版本的，还是cuda版本的原因？

DayBreak-u avatar Nov 29 '19 13:11 DayBreak-u

你好，你用的torch是什么版本的，还是cuda版本的原因？

我的版本是pytorch1.3.1 cudatoolkit 10.1.243

harryQAQ avatar Nov 29 '19 14:11 harryQAQ

我是1.2.0 cuda 10.1 应该没问题的呀

DayBreak-u avatar Nov 30 '19 03:11 DayBreak-u

我是1.2.0 cuda 10.1 应该没问题的呀

开了个干净的虚拟环境，保持版本一致pytorch 1.2.0、torchvision 0.4，问题解决了，多谢~

harryQAQ avatar Dec 02 '19 02:12 harryQAQ