deep-hough-transform icon indicating copy to clipboard operation
deep-hough-transform copied to clipboard
verified publisher icon Hanqer

Importing problem

Open dymwan opened this issue 3 years ago • 1 comments

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: image

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

dymwan avatar Sep 25 '22 04:09 dymwan

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

Hanqer avatar Oct 08 '22 06:10 Hanqer