JianqiangWan / Super-BPD

Super-BPD: Super Boundary-to-Pixel Direction for Fast Image Segmentation (CVPR 2020)
Apache License 2.0
200 stars 37 forks source link

Failed to compile the BPD module with errors under both Pytorch1.2 & Pytorch1.5 #2

Closed PkuRainBow closed 4 years ago

PkuRainBow commented 4 years ago

Hi, Really nice work!

We have tried to compile the BPD module in the post_process folder and got different error information based on Pytorch1.2 or Pytorch1.5 as below.

image

Thanks!

JianqiangWan commented 4 years ago

Please use pytorch >= 1.3.0. For pytorch = 1.5.0, it looks like a compatibility issue with gcc and nvcc , upgraded to gcc 7 and the compilation passed without a problem. ref. https://github.com/rusty1s/pytorch_geometric/issues/942.