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

cuda support for docker container #140

Open
rejuce opened this issue Dec 30, 2023 · 5 comments · Fixed by #170
Open

cuda support for docker container #140

rejuce opened this issue Dec 30, 2023 · 5 comments · Fixed by #170
Assignees
Labels
bug Something isn't working

Comments

@rejuce
Copy link
Contributor

rejuce commented Dec 30, 2023

i have cuda, my wsl envirnment reconises it , but i suppose the docker container from this project does not yet support cuda?

(there are no cuda toolkit commands available and also epub2tts does not run with cuda)

I entered now the container over bash an running directly /opt/epub2tts.py

@aedocw
Copy link
Owner

aedocw commented Dec 30, 2023

Interesting, I have not looked at the docker/cuda/windows combo, apologies for that. I'll try to spend some time this week poking at it.

@aedocw aedocw added the bug Something isn't working label Dec 30, 2023
@rejuce
Copy link
Contributor Author

rejuce commented Dec 30, 2023

the default way for doing this seems to be to derive the image from nvidia/cuda:base,, my wsl environment / docker environment supports CUDA as I have other containers running that can use it (eg jupyterlab). but the base image epuc2tts derives from, does not include CUDA stuff

@aedocw
Copy link
Owner

aedocw commented Dec 31, 2023

In addition to my updating the Dockerfile to use an image with native GPU support, anyone wanting to run the container with CUDA support will need to install the NVIDIA Container Toolkit:

WARNING: The NVIDIA Driver was not detected.  GPU functionality will not be available.
   Use the NVIDIA Container Toolkit to start this container with GPU support; see
   https://docs.nvidia.com/datacenter/cloud-native/ .

@aedocw
Copy link
Owner

aedocw commented Dec 31, 2023

Need to determine whether the image with CUDA support will run fine in environment without CUDA, or if we need to create a specific GPU/CUDA image and one without GPU support.

@aedocw aedocw self-assigned this Jan 7, 2024
@aedocw aedocw linked a pull request Jan 7, 2024 that will close this issue
@aedocw aedocw reopened this Jan 7, 2024
@aedocw
Copy link
Owner

aedocw commented Jan 7, 2024

There is now a cuda12 container "docker pull ghcr.io/aedocw/epub2tts:release-cuda12", though I tested with the following and it does not detect the GPU:

alias cudaepub='docker run --gpus=all -e COQUI_TOS_AGREED=1 -v "$PWD:$PWD" -v ~/.local/share/tts:/root/.local/share/tts -w "$PWD" ghcr.io/aedocw/epub2tts:release-cuda12'
cudaepub testing.txt --engine xtts --speaker "Damien Black"

I will poke around and see what else I need to install, though at this point I think that container should work properly and support CUDA GPU.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug Something isn't working
Projects
None yet
Development

Successfully merging a pull request may close this issue.

2 participants