Looking for torchaudio compatible NGC container

Dear community,

I’ve been working on LLM finetuning on the NGC container and look to integrate TTS tools in the setup, unfortunately torchaudio runs into errors if I try to build it from source.

ERROR: No matching distribution found for torch==2.4.0a0+07cecf4

I also get errors with regular pip install torch==2.4.0 (I assume torch expects identitical model name and number to run smoothly)

How to move forward from here? I guess I can try other NGC container variants with torchaudio - or - I am curious to hear what other NVIDIA toolkit (CuDNN) enabled containers I should use for working with huggingface’s transformers (bitsandbytes) along with torchaudio so please, give me your best tips!

This is my first ever post on the forum, be gentle :-)