Running on an nvidia Blackwell GPU

[enamigi]

First I had installed Audiocraft_plus on ubuntu 24.04 under wsl with an Ampere GPU (rtx 3060). It worked with the instructions on the repository.

Later I upgraded to a blackerll GPU (RTX 5060 TI). The old installation didn't work. I tried updating to the latest pytorch and cuda. Bad decision. Didn't worked. I eventually got it to work but the quality was awful.

I deleted the old environment and remembered the instructions say:

"AudioCraft requires Python 3.9, PyTorch 2.0.0. To install AudioCraft, you can run the following..."

I thought that maybe Audiocraft_plus can run in the next newer python version. I use anaconda and created a new environment.

conda create -n ap_cu128 python=3.10

This will create an environment with python 3.10.

Then Installed the lowest version of pytorch that supports cuda 12.8, the first version with support for blackwell generation GPUs.

conda activate ap_cu128

pip install torch==2.7.0 torchvision==0.22.0 torchaudio==2.7.0 --index-url https://download.pytorch.org/whl/cu128

I needed to install dotenv

pip install python-dotenv

Then i installed via pip

pip install -e .

This will install AudioCraft_Plus completely. It finished without error or warning.
When I launched the gradio app, I got an error that some method is missing. The installation process gets a very new version (like 6.2 or something). I downgraded gradio to verision 3.50.0

pip install gradio==3.50.0

This operation replaced the newer version with a legacy one that works. The quality is again good. No strange noises in the music.

I hope this is useful for somebody.