m-bain / whisperX

WhisperX: Automatic Speech Recognition with Word-level Timestamps (& Diarization)
BSD 2-Clause "Simplified" License
12.52k stars 1.32k forks source link

libtorch_cuda.so error #471

Open AlexandderGorodetski opened 1 year ago

AlexandderGorodetski commented 1 year ago

Guys,

I get following error during whisperx importing

OSError: libtorch_cuda.so: cannot open shared object file: No such file or directory

I do not have this error when I work on raw machine.

I get this error only when I work from docker.

But I prefer to work from docker.

Any help is appreciated.

Thanks a lot, AlexG.

sorgfresser commented 1 year ago

You need specific container runtimes when running pytorch in docker, e.g. the nvidia container toolkit. Maybe that's an issue here?