cherubicXN / hawp

Holistically-Attracted Wireframe Parsing [TPAMI'23] & [CVPR' 20]
MIT License
298 stars 52 forks source link

python setup.py #13

Closed sunhd96 closed 3 years ago

sunhd96 commented 3 years ago

Hello, sorry to bother you via issue creation. I would like to ask you about compiling cuda files, python setup.py build_ext --inplace. First I compile in cuda11.0 and it fails. Then compile in cuda10.0 to get the _C.***.so file. Then move it to the parsing directory, run the train.py file and still report an error. Traceback (most recent call last): File "train.py", line 8, in from parsing.detector import WireframeDetector File "/home/sun/line/hawp-master/parsing/detector.py", line 4, in from parsing.encoder.hafm import HAFMencoder File "/home/sun/line/hawp-master/parsing/encoder/init.py", line 1, in from .hafm import HAFMencoder File "/home/sun/line/hawp-master/parsing/encoder/hafm.py", line 4, in from parsing import _C ImportError: cannot import name '_C' from 'parsing' (/home/sun/line/hawp-master/parsing/init.py) Thanks, Hao Sun

cherubicXN commented 3 years ago

Hi Hao Sun,

Sorry to reply to you late. I just saw a new issue that was created 12 days before.

Could you directly compile it with CUDA10.0 in the project directory?

sunhd96 commented 3 years ago

Hi Hao Sun,

Sorry to reply to you late. I just saw a new issue that was created 12 days before.

Could you directly compile it with CUDA10.0 in the project directory?

Thanks for your reply. According to your suggestion, I used cuda10.1 to compile directly and it succeeded. thank you very much

cherubicXN commented 3 years ago

That's great.