Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Move DropoutLayer random number generator to GPU #7

Closed
danielsuo opened this issue Nov 18, 2015 · 1 comment
Closed

Move DropoutLayer random number generator to GPU #7

danielsuo opened this issue Nov 18, 2015 · 1 comment

Comments

@danielsuo
Copy link
Contributor

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).

@vinjn
Copy link

vinjn commented Nov 19, 2015

cuRAND might be a good candidate.
https://developer.nvidia.com/curand

@danielsuo danielsuo modified the milestone: v0.2.0 Release Nov 21, 2015
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants