Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

No module named 'custom_conv' #3

Open
stefenmax opened this issue Apr 15, 2021 · 4 comments
Open

No module named 'custom_conv' #3

stefenmax opened this issue Apr 15, 2021 · 4 comments

Comments

@stefenmax
Copy link

When i used setup.py it raised:
File "setup.py", line 26, in
import custom_conv
ModuleNotFoundError: No module named 'custom_conv'
how can i fix it thanks!

@teboli
Copy link
Owner

teboli commented Apr 21, 2021

Hi! This means that the cuda module has not compiled on your machine. However, it is only necessary if you intend to use the non-uniform model. If not, you can skip running setup.py and comment https://github.com/teboli/CPCR/blob/master/layers/functions.py#L7 (import custom_conv in this case).

I will strive to fix this bug ASAP, sorry for inconvenience.

@stefenmax
Copy link
Author

OK Thanks, it works. But does it mean if the cuda has not compiled on my machine, the code can't using the GPU.

@stefenmax
Copy link
Author

By the way, the compile is failed cause

subprocess.CalledProcessError: Command '['which', 'g++']' returned non-zero exit status 1.

@teboli
Copy link
Owner

teboli commented Apr 22, 2021

OK Thanks, it works. But does it mean if the cuda has not compiled on my machine, the code can't using the GPU.

This is true only for my custon non-uniform convolution layer whose cuda code was shipped with the rest of the project. However compiling it is machine-dependent and I must think about a way to fix it. On the other hand, the code for uniform deblurring only rely on Pytorch implementation and thus is already compiled for your specific machine; you can run the method for uniform deblurring on your own GPU.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants