Implementation on V100

Question

GewelsJI opened this issue 2 years ago · 4 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?

Answer 1 · 2022-08-02T05:21:21.000Z

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

Answer 2 · 2022-08-02T23:23:02.000Z

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

Answer 3 · 2022-10-20T04:38:43.000Z

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

Answer 4 · 2022-10-20T14:47:52.000Z

@StevenGrove Thank you again.