seanghay / uvr-mdx-infer

Ultimate Vocal Remover Inference CLI
33 stars 6 forks source link

Not using GPU #2

Closed deletedcookie closed 6 months ago

deletedcookie commented 6 months ago

It is only using my CPU and not my GPU. Do I have to change anything to make it use my GPU?

I only have a 1050ti, which is below the minimum specs, but it works fine on the GUI version and is twice as fast as my CPU.

seanghay commented 6 months ago

It will use the GPU when possible. However, if it doesn't you might need to check your PyTorch dependencies to make sure it's the cuda version not cpu version. To install PyTorch CUDA, you may need to follow: https://pytorch.org/get-started/locally/

deletedcookie commented 6 months ago

Thanks, I thought I had the cuda version but it didn't for some reason. It works fine now after uninstalling and reinstalling with the cuda version.