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

How to use GPU rather than CPU #698

Open
SoftologyPro opened this issue May 14, 2024 · 0 comments
Open

How to use GPU rather than CPU #698

SoftologyPro opened this issue May 14, 2024 · 0 comments
Labels
question Further information is requested

Comments

@SoftologyPro
Copy link

Hello,

Trying the first example to separate an mp3. ie

from asteroid.models import BaseModel
import soundfile as sf

model = BaseModel.from_pretrained("mpariente/DPRNNTasNet-ks2_WHAM_sepclean")
model.separate("music.mp3",resample=True)

It worked (well worked as in just made 2 distorted versions of the source music without separating anything it seems) but it uses 100% CPU and seems to ignore the GPU.
How do I tell it to use the GPU?
Thanks.

@SoftologyPro SoftologyPro added the question Further information is requested label May 14, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
question Further information is requested
Projects
None yet
Development

No branches or pull requests

1 participant