fatal error: torch/extension.h: No such file or directory

jackroos / VL-BERT

Code for ICLR 2020 paper "VL-BERT: Pre-training of Generic Visual-Linguistic Representations".

MIT License

738 stars 110 forks source link

fatal error: torch/extension.h: No such file or directory #64

Closed whqwill closed 3 years ago

whqwill commented 3 years ago

When I run ./scripts/init.sh, it shows: /opt/mnt/haiqing/Project/VL-BERT-HAKE/common/lib/roi_pooling/cpu/vision.h:3:29: fatal error: torch/extension.h: No such file or directory How to solve it?

alice-cool commented 3 years ago

Could you solve it , I find the same error?

Jiejiegary commented 3 years ago

@whqwill Hi, May I know how did you fix the issue? My Pytorch version is 1.4 but I still have this bug.