GPU Acceleration?

[lim-xem]

Hi!

Thanks for this, especially the sample code. I'm new here and this has been a great introduction to EEG and deep neural networks.

Does this code tap on the GPU, e.g. via CUDA/cuDNNs? I have a mid-range Nvidia GPU on my laptop, and when running previous deep learning programmes using PyTorch on a Jupyter Notebook, I have been able to successfully speed up training times by about 5 times using CUDA.

It doesn't seem like this code does, but I'd just like to check with you because I don't have much experience! Thanks.

Best regards,
Michelle

[lim-xem]

I've figured it out. If we set up Theano=1.0 (not 0.8.2) and CUDA drivers etc according to the Theano documentation, i.e. differently from what's indicated in the README, GPU acceleration is automatic!