Move DropoutLayer random number generator to GPU
danielsuo opened this issue · comments
Daniel Suo commented
Currently we rely on the CPU for random numbers. To make this slightly less painfully slow, we create a new thread, but this is slower than using the GPU and also causes problems in Windows (#4).
Jing Zhang commented
cuRAND might be a good candidate.
https://developer.nvidia.com/curand