search

zsef123 / Connected_components_PyTorch

A PyTorch implementation of Connected Components Labeling
BSD 3-Clause "New" or "Revised" License
105 stars 24 forks source link

update for new cuda? #9

Open kechanf opened 6 months ago

kechanf commented 6 months ago

Hello,

I am interested in updating your tool to ensure its compatibility with the programming environment expected in 2024, including support for CUDA 11.8. Could you provide guidance on the necessary modifications or steps required to achieve this?

Thank you for your time and support.

arunthurai commented 5 months ago

Hi, I was also looking to implement the connected components count as part of a loss function for a segmentation task in PyTorch but was having issues compiling with CUDA version 12.2. Was there updates to this/could I receive some support for this. Thank you!