Giters

pbashivan / EEGLearn

A set of functions for supervised feature learning/classification of mental states from EEG based on "EEG images" idea.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

GPU Acceleration?

lim-xem opened this issue · comments

commented

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

commented

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!