search

ZFTurbo / Music-Source-Separation-Training

Repository for training models for music source separation.
MIT License
491 stars 68 forks source link

CPU vs GPU #93

Closed AirJCovers34 closed 1 week ago

AirJCovers34 commented 1 week ago

How can I specify for the program to use the GPU instead of the CPU?

image

ZFTurbo commented 1 week ago

Code automatically uses CUDA if it's avialable.

AirJCovers34 commented 1 week ago

Ok. That was my mistake: My Cuda wasn’t installed in the venv I’m using to run the script. Thank you.