cerndb / dist-keras

Distributed Deep Learning, with a focus on distributed training, using Keras and Apache Spark.
http://joerihermans.com/work/distributed-keras/
GNU General Public License v3.0
623 stars 169 forks source link

GPU-compatible release #49

Open smurching opened 6 years ago

smurching commented 6 years ago

The current dist-keras release seems to depend on tensorflow (see setup.py). Would it be possible to make a release (on PyPi) that depends on tensorflow-gpu? Thanks!