pyannote uninstalling torch cuda and install torch cpu

[versus666jzx]

Tested versions

• Reproducible in Version 3.1.1

System information

win11, pyannote.audio 3.1.1

Issue description

If I install pyannote.audio in an env in which torch with cuda is already installed, then during installation the torch with cuda is removed and torch cpu is installed. Even though the torch versions satisfy requirements.txt.

Minimal reproduction example (MRE)

pip install torch torchaudio --index-url https://download.pytorch.org/whl/cu118
pip install pyannote.audio

[Valeronich]

Same problem omg

[gdk00]

Struggled with that too...

[beton18]

same issue

[hbredin]

I'd suggest using the same --index-url when installing pyannote.audio.

pip install pyannote.audio --index-url https://download.pytorch.org/whl/cu118

[versus666jzx]

I'd suggest using the same --index-url when installing pyannote.audio.

pip install pyannote.audio --index-url https://download.pytorch.org/whl/cu118

Expectedly doesn't work.

[versus666jzx]

But if use this

pip install pyannote.audio --extra-index-url https://download.pytorch.org/whl/cu118

then torch and torchaudio packages with cuda support will be installed

It would be cool if this was somehow reflected in the installation documentation or if the requirements.txt or setup files were corrected.

[hbredin]

It would be cool if you could open a PR fixing this now that you have found a solution.