beeldengeluid / dane-whisper-asr-worker

MIT License
2 stars 0 forks source link

22 gpu docker #26

Closed greenw0lf closed 4 months ago

greenw0lf commented 4 months ago

Closes #22

Finally made it work with the GPU also. Only had to change the base image to an official one from NVIDIA (which also supports cuDNN 8).

I added a description on how to set it up for a different worker and run it on the wiki of the example worker and pointed to the wiki in the README.

Veldhoen commented 4 months ago

Not really related to GPU-containerization, but maybe add config.yml to .dockerignore, to make sure that the worker is using the config mounted in /root/.DANE

greenw0lf commented 4 months ago

It already is there (/config)

greenw0lf commented 4 months ago

Shall I merge?

greenw0lf commented 4 months ago

@Veldhoen reminder that I left some comments. Shall I go ahead and merge?

Veldhoen commented 4 months ago

Sorry for the late reply - yes go ahead!