Importing problem

[dymwan]

Hi,
Firstly, your fantastic work is really helpful. And there is a small problem with using:

After importing the torch and deep_hough package in python, the error
imtens = imtens.cuda()
RuntimeError: CUDA error: unrecognized error code

occurred when I try to allocate a tensor from CPU to GPU via tensor.cuda().
However, if I just simply do not import deep_hough, the error disappeared. I am sure that my software was set ok.

Here is a small demo:

My software setup is:
pytorch 1.7 + cu11
Waiting for some suggestions sincerely.

[Hanqer]

I think this may caused by the incompatible cuda version, you can try to compile it using cuda10.