search

kexinyi / ns-vqa

Neural-symbolic visual question answering
258 stars 61 forks source link

no attribute 'roi_align_forward_cuda' #2

Closed Oscar860601 closed 5 years ago

Oscar860601 commented 5 years ago

Hi, I encounter this error message module 'modeling.roi_xfrom.roi_align._ext.roi_align' has no attribute 'roi_align_forward_cuda' both running pretrain scene parser and training scene parser. Do you have solution of this?

Thanks.

My envs setting is below: python=3.5 pytorch=0.4.0 CUDA=9.0

Oscar860601 commented 5 years ago

I accidentally solve this by myself. I think changing the CUDA_PATH in maks.sh and assigning at least one GPU for make.sh can solve this.