search

szilard / benchm-dl

Playing with various deep learning tools and network architectures
69 stars 11 forks source link

H2O's GPU Deep Learning #1

Open arnocandel opened 7 years ago

arnocandel commented 7 years ago

Sneak preview for H2O's Deep Water project: https://github.com/h2oai/h2o-3/blob/master/examples/deeplearning/notebooks/deeplearning_benchmark_mnist.ipynb

To try this script yourself on Ubuntu 16 with CUDA8, CUDNN5, simply pip install this package: https://slack-files.com/T0329MHH6-F2TN01TUN-b0577b68da

You will also want to install the matching mxnet egg: https://slack-files.com/T0329MHH6-F2PU85GEN-2f4fee68e2

More instructions and links provided here: https://github.com/h2oai/deepwater

Please note that this is not a final version for benchmark purposes, but rather a community preview among friends.

g2.2xlarge: 47s p2.2xlarge: TBD 2x 10-core Xeon with GTX1080: 26s

Cheers, Arno

szilard commented 7 years ago

Awesome, and thanks for comments @arnocandel. For now I'm just playing around, I was trying to keep this repo quiet LOL. I've been actually playing with mxnet as well, will need to add that to the repo. Really looking forward for deep water to come out GA - might also try it before that, thanks for the instructions above.