TreeFilter-Torch Implementation on V100

Implementation on V100

Open GewelsJI opened this issue 1 year ago • 2 comments

Hi,

Thank you for providing such a nice paper. I have a question here. The default setting indicates the implementation on pytorch1.2.

But, if possible, could this library (TreeFilter-Torch/furnace/kernels/lib_tree_filter) be implemented on V100 GPU? Are there any modifications?

Aug 02 '22 01:08 GewelsJI

Yes, it can run on V100 GPU. What issues did you have when compiling the code?

Aug 02 '22 05:08 StevenGrove

I guess the issue coming from the CUDA version. What's your version of CUDA and pytorch on V100?

Aug 02 '22 23:08 GewelsJI

I have updated the code. Now it can support cuda 11 and up to pytorch 1.12.1.

Oct 20 '22 04:10 StevenGrove

@StevenGrove Thank you again.

Oct 20 '22 14:10 GewelsJI