pytorch problem: /_roi_align.so: undefined symbol: PyInt_FromLong

Question

pytorch problem: /_roi_align.so: undefined symbol: PyInt_FromLong

ccczhixian opened this issue 5 years ago · 9 comments

Hi , when I run the demo_eval.py on linux, there had an error:
ImportError :/pytorch_cropping_evaluation/roi_align/_ext/roi_align/_roi_align.so: undefined symbol: PyInt_FromLong

could you please tell me how can I solve the problem?

Answer 1 · 2019-07-05T07:54:51.000Z

Hi , when I run the demo_eval.py on linux, there had an error:
ImportError :/pytorch_cropping_evaluation/roi_align/_ext/roi_align/_roi_align.so: undefined symbol: PyInt_FromLong

could you please tell me how can I solve the problem?

You need to compile the source code of roi_align in your environment and generate the .so file.
Please see this issue: #1

Answer 2 · 2019-07-05T08:22:58.000Z

sm_52在哪里定义

Answer 3 · 2019-07-05T08:24:39.000Z

sm_52在哪里定义

这个是什么？

Answer 4 · 2019-07-05T08:25:48.000Z

ccczhixian commented 5 years ago

Answer 5 · 2019-07-05T08:27:21.000Z

我编译的时候报错

Answer 6 · 2019-07-05T08:29:46.000Z

这个参数好像和cuda版本有关。

Answer 7 · 2019-07-05T08:47:59.000Z

我编译的时候报错

我觉得和CUDA版本以及使用的显卡都有关系。
可以联系一下工作站/服务器的管理员，或者找amax的销售也可以，确认一下当前的cuda版本以及所使用的显卡型号，根据显卡的型号，从下面的链接里面找出对应的gpu architecture的架构号码，修改sm_52。通常1080或者1080ti都是sm_60或以上，2080ti等的sm号码更大一些，sm_52适应于比较早的显卡结构。
https://arnon.dk/matching-sm-architectures-arch-and-gencode-for-various-nvidia-cards/

Answer 8 · 2019-07-10T09:11:40.000Z

怎么在Ubuntu下重新编译roi_align源码生成.so文件啊？

Answer 9 · 2019-07-10T09:14:23.000Z

怎么在Ubuntu下重新编译roi_align源码生成.so文件啊？

相应的文件夹下给了make.sh文件，不过好像对pytorch的版本有一定的要求，pytorch0.4.1可以，太新的版本比如1.0之后好像改了一些底层的东西，就会报错。