Pre-trained usage with CPU / without Nvidia GPU

resemble-ai / resemble-enhance

AI powered speech denoising and enhancement

https://huggingface.co/spaces/ResembleAI/resemble-enhance

MIT License

1.09k stars 103 forks source link

Pre-trained usage with CPU / without Nvidia GPU #16

Closed Hahlh closed 4 months ago

Hahlh commented 5 months ago

Is there any chance of enabling the usage of the pre-trained model with a CPU / non-Nvidia GPU? (-> RuntimeError: Found no NVIDIA driver on your system)

Or is there no hope due to cuda?

Thank you for the awesome work!

4lvrz commented 4 months ago

At least inference is posible over CPU :)

Hahlh commented 4 months ago

@4lvrz Oh, I actually missed the --device cpu flag and the automatic switch always failed for me with:

  File "/home/user/.local/share/pipx/venvs/resemble-enhance/lib/python3.11/site-packages/torch/cuda/__init__.py", line 298, in _lazy_init
    torch._C._cuda_init()
RuntimeError: Found no NVIDIA driver on your system. Please check that you have an NVIDIA GPU and installed a driver from http://www.nvidia.com/Download/index.aspx

Thank you for making me have another look.