Does this work on AMD GPUs?

facebookresearch / audiocraft

Audiocraft is a library for audio processing and generation with deep learning. It features the state-of-the-art EnCodec audio compressor / tokenizer, along with MusicGen, a simple and controllable music generation LM with textual and melodic conditioning.

MIT License

20.14k stars 2.01k forks source link

Does this work on AMD GPUs? #103

Open jthompson05 opened 1 year ago

jthompson05 commented 1 year ago

Hello. I am here to ask if there is a way to get this project working on AMD GPUs. I have tried it via https://github.com/sdbds/audiocraft-webui but ran into an error with my gpu. I have the Radeon RX 6700S.

sdbds commented 1 year ago

now it use NVIDIA CUDA for default, i dont have AMD GPU so i cant ensure ROCm can be used. or you can use as MAC M1 with CPU

PolyVector commented 1 year ago

I think the main issue is that this project relies on xformers, so that breaks ROCm compatibility.

Neoony commented 11 months ago

hopefully soon this stuff will start changing https://www.phoronix.com/news/AMD-HIP-SDK-Windows

danielfl commented 2 months ago

yes, it works if you install the torch version released by AMD. example: python3 -m pip install torch==2.3.0.dev20240301+rocm5.7 --index-url https://download.pytorch.org/whl/nightly/rocm5.7

and so on..