ckkelvinchan / RealBasicVSR

Official repository of "Investigating Tradeoffs in Real-World Video Super-Resolution"
Apache License 2.0
906 stars 136 forks source link

ERROR: Failed building wheel for mmcv-full #29

Closed AK391 closed 2 years ago

AK391 commented 2 years ago

when using

mim install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cpu/torch1.10.0/index.html

I am looking to setup a space on huggingface spaces(https://huggingface.co/spaces) for this model at https://huggingface.co/spaces/akhaliq/RealBasicVSR

I was able to get the model working in colab but the space does not support cuda, is there a way around this thanks?

space code: https://huggingface.co/spaces/akhaliq/RealBasicVSR/blob/main/app.py#L4

ckkelvinchan commented 2 years ago

Hello, thank you for your efforts to set up a space for RealBasicVSR.

mim will automatically determine the appropriate version based on your pytorch and cuda version. Therefore I think simply mim install mmcv-full should be fine. Could you have a try and let me know if it works?

Thanks again.

ckkelvinchan commented 2 years ago

Oh my bad. Do you mean huggingface does not support cuda?

AK391 commented 2 years ago

@ckkelvinchan Each Spaces environment is limited to 16GB RAM and 8 CPU cores https://huggingface.co/docs/hub/spaces

ckkelvinchan commented 2 years ago

@AK391 mmcv-full can only be installed with CUDA. If CUDA is not available, you can use mim install mmcv. It lacks CUDA ops but it shouldn't affect RealBasicVSR. Please have a try.

AK391 commented 2 years ago

@ckkelvinchan thanks, I tried that now getting

Traceback (most recent call last): File "inference_realbasicvsr.py", line 10, in from mmedit.core import tensor2img ModuleNotFoundError: No module named 'mmedit'

mmedit is installed here https://huggingface.co/spaces/akhaliq/RealBasicVSR/blob/main/app.py#L4

ckkelvinchan commented 2 years ago

@AK391 Thank you for your feedback. Please try the following:

  1. pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cpu/torch1.10/index.html
  2. pip install mmedit

I tried and it works in colab. Please let me know if it still does not work.

CleanShot 2022-03-12 at 08 23 07@2x
AK391 commented 2 years ago

@ckkelvinchan thanks it is working now, opened a PR here https://github.com/ckkelvinchan/RealBasicVSR/pull/30