How to enable GPU or CPU?

[juangea]

Hi there.

How can I define the device to use? I'm using it as a library with Python.

[alcoftTAO]

vc = VC()

vc.config.use_cuda()     # For GPU using CUDA
vc.config.use_cpu()      # For CPU

vc.get_vc(model_path)

I've also tested this code on AMD GPUs and it doesn't work, make sure you have an NVIDIA GPU and you have it's drivers and CUDA installed.

[juangea]

Thanks!

I also noticed that it automatically uses the GPU if Torch for CUDA is installed, I was unsure about it being installed or not in this project, that's why I was unsure about the GPU.

But your answer is very useful! :D