GPU acceleration on Windows

If you have graphic cards produced by NVIDIA with more than 6GiB of memory, you can separate tracks with GPU acceleration. To achieve this, you must install Pytorch with CUDA. If Pytorch was already installed (you already installed Demucs for instance), first run python3 -m pip uninstall torch torchaudio. Then visit the PyTorch Home Page and follow the guide to install Pytorch with CUDA support. You can also refer to the Demucs Docs.

Then you can use the -d or --device option in Stemgen to use CUDA.

python3 stemgen.py track.wav -d cuda