search

megvii-model / HINet

Other
367 stars 62 forks source link

ImportError #38

Open Yuri6037 opened 2 years ago

Yuri6037 commented 2 years ago

Your network does not run on my research machine:

Traceback (most recent call last):
  File "basicsr/demo.py", line 10, in <module>
    from basicsr.models import create_model
ImportError: cannot import name 'create_model'

EDIT: Just tried on my local computer (RTX 3060) different error:

Traceback (most recent call last):
  File "/home/X/HINet/basicsr/demo.py", line 10, in <module>
    from basicsr.models import create_model
ModuleNotFoundError: No module named 'basicsr'
mexthecat commented 2 years ago

maybe you have forgotten: "python setup.py develop --no_cuda_ext"

Yuri6037 commented 2 years ago

I've got the software to run on my RTX 3060 by moving main.py outside of the basicsr directory and by implementing image decomposition to avoid VRAM overload (my RTX only has 6Gb of VRAM).

On the remote machine according to some Python Matrix channel it's unfixable: the problem is because you've named the directory basicsr which conflicts with the module I have installed (also named basicsr).

mayorx commented 2 years ago

Hi, Yuri6037,

Thanks, and you could try export PYTHONPATH=path_to_the_HINet_folder in your bash before run basicsr/demo.py. It might be helpful.

dante1024 commented 2 years ago

maybe you have forgotten: "python setup.py develop --no_cuda_ext"

What is the operation of "python setup.py develop --no_cuda_ext" used for?