RTX- 5090 (sm_120) for F5-TTS

[musselee]

Does any body successfully use RTX- 5090 (sm_120) to apply F5-TTS?
The instruction to setup env:
"pip install torch==2.4.0+cu124 torchaudio==2.4.0+cu124 --extra-index-url https://download.pytorch.org/whl/cu124" does not support RTX- 5090 (sm_120)!

[dpp-user]

you need nightly build cuda 12.8 and torch 2.7.0 if i remember correctly , im working with the 5090 with it.

[tanvan81]

can you write more detail ? Thanks

[dpp-user]

i dont remember the exact command, but just ask chat gpt how to install cuda 12.8 and torch 2.7.0 or later all nightly build

[JasonOdinberg]

I am using this for the 6000 pro (should work too on 5090) :
pip install torch==2.9.0+cu128 torchaudio==2.9.0+cu128 --extra-index-url https://download.pytorch.org/whl/cu128
pip install torchcodec==0.9.1 (uninstall old one first -> pip uninstall torchcodec)

This is with the latest ffmpeg btw.

[gwnotinga]

Thanks JasonOdinberg. I can confirm that your commands work on 5090.
Much appreciated.