I’m running Whisper and Pyannote Audio diarization in a google cloud VM instance group. Whisper is able to detect my GPU just fine, but Pyannote cannot. torch.cuda.is_available() returns true.
Below describes my setup, which should make the issue reproducible. I have tried to use the development branch, torch 2.0.0, upgraded cuda, etc. Note: Pyannote worked for me in the past, not sure what happened.
Running the following txt file:
RUN apt-get update && apt-get install -y --no-install-recommends
apt-get clean &&
rm -rf /var/lib/apt/lists/*
Followed these directions for starting the VM with GPU: