search

youngwanLEE / centermask2

[CVPR 2020] CenterMask : Real-time Anchor-Free Instance Segmentation
Other
772 stars 159 forks source link

CenterMask2 with pytorch > 1.6 doesn't work #77

Closed ApelSYN closed 3 years ago

ApelSYN commented 3 years ago

In autumn, new GPUs from NVIDIA 30xx series were released, which only work on CUDA> = 11.xx. In order for GPU 30xx to work, you need to install pytorch> = 1.7. Detectron2 will work fine with pytorch 1.7, CenterMask will not work with new pytorch.

Are you planning to support new torch versions?

youngwanLEE commented 3 years ago

@ApelSYN Thanks for your attention.

I have a plan to update this repo for compatibility with the latest detectron2 & CUDA versions in the near future.

youngwanLEE commented 3 years ago

@ApelSYN

Could you detail your problem or log?

youngwanLEE commented 3 years ago

@ApelSYN I update the code for compatibility with pytorch1.7 and the latest detectron2.

Please re-open this issue if you encounter this problem again.

ApelSYN commented 3 years ago

Now all working fine. Thanks.