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

Is there any possible to use multi GPU to inside the compute_and_store_features_batch function #1331

Open
OswaldoBornemann opened this issue May 7, 2024 · 3 comments

Comments

@OswaldoBornemann
Copy link

So suppose I have 4 million audio files, and I want to extract the features using Encodec. I found that the extracting process is kind of slow, where Computing features in batches: 183301it [19:36, 176.40it/s]. The extracting process only leverages single GPU. So I was wonder whether we can use multi-GPU to accelerate the extraction process in the compute_and_store_features_batch.

@pzelasko
Copy link
Collaborator

pzelasko commented May 7, 2024

It will be more robust if you split your manifest into parts and process each part separately. You can launch the script multiple times with GPU ID as an argument.

@OswaldoBornemann
Copy link
Author

I see. Thank you.

@OswaldoBornemann
Copy link
Author

So how to split the manifest like the cut file did?

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