Can I use cuda to speed up the training process?

Question

Can I use cuda to speed up the training process?

Hao-Yuan-He opened this issue a year ago · 1 comments

I'm wondering if I can use Cuda to speed up the training process, the example/MNIST/addition.py seems to only use CPU.

Answer 1 · 2023-11-20T21:09:37.000Z

Hi

You can put the neural networks behind the neural predicate on the GPU to accelerate these by callign the .cuda(device) method. The other parts of the inference cannot be accelerated.

Kind regards,
Robin