Support for cuda 10.2

[xiyangl3]

Hello,

Can you update support with cuda 10.2?

Thanks

[tridao]

Are you having issue with the Python part of the code, or the C++/CUDA part?
The Python part should work with Pytorch 1.0 (I'm not sure about newer Pytorch versions).
Which Pytorch version are you using?

[xiyangl3]

Hi,

I am using pytorch 1.6 with cuda version 10.2. I have tested pytorch 1.4 + cuda 10.1, which works well. But I am having some trouble installing with cuda 10.2 now.

Thanks.

[tridao]

Do you mean installing the CUDA extensions with python setup.py install?

[xiyangl3]

Yes

[tridao]

I've just pushed a commit to update the CUDA extensions. Can you try them now?