CUDA

Hi,

Thanks for the implementation. Can I write the training without CUDA? I keep getting an assertion error that I do not have CUDA (I'd rather not to install the drivers required).

Thanks,
Arik.