search

pbashivan / EEGLearn

A set of functions for supervised feature learning/classification of mental states from EEG based on "EEG images" idea.
GNU General Public License v2.0
710 stars 220 forks source link

GPU Acceleration? #32

Closed lim-xem closed 6 years ago

lim-xem commented 6 years ago

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 commented 6 years ago

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!