What would need to be done to utilize gpu? about docker-facebook-demucs HOT 6 CLOSED

I'll take a look because it would be great to run the demucs in different modes 💪🏼

from docker-facebook-demucs.

any branch that might have been started that i can contribute to?

from docker-facebook-demucs.

Hi @CptanPanic ,

I guess we could add a Dockerfile that starts from the Nvidia image. Do you have the Nvidia image you're referencing?

Thanks for your contribution :)

from docker-facebook-demucs.

Maybe something like nvidia's pytorch image? nvcr.io/nvidia/pytorch:latest as I think demucs uses pytorch.

from docker-facebook-demucs.

@aavetis Thanks for contributing to the project! 🚀 I started the development in my local environment but didn't get an early version that worked, I got this error building the image nvcr.io/nvidia/pytorch:22.08-py3:

stderr: nvidia-container-cli: initialization error: load library failed: libnvidia-ml.so.1: cannot open shared object file: no such file or directory: unknown

It can be related to this issue NVIDIA/libnvidia-container#135 but I've not digged into the issue itself yet.

So great to see a pull request with a first approach on that :) I'll take a look and we can discuss there the changes! Thanks again and welcome to the docker-facebook-demucs repo!

from docker-facebook-demucs.

Closing as we've merged the support for GPU as well as arm64 and some extra options thanks to @aavetis and @chucknelson for their contributions #22 #24 !

from docker-facebook-demucs.