-
Notifications
You must be signed in to change notification settings - Fork 4
Conversation
Update jupyterhub and add Jupyter Notebook Deep Learning Stack
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Hello @AyadiAmen,
Is this Docker image sufficient to run Tensorflow? Should not this image instead be coupled with a ML engine for Tensorflow (a bit like Spark) ? @titsitits?
@alexnuttinck I think it is, Tensorflow is a library for dataflow and differentiable programming used to create models and that image is sufficient to do so, I tested: https://www.tensorflow.org/tutorials/quickstart/beginner & https://www.tensorflow.org/tutorials/load_data/csv. @titsitits what do you think ? |
Notebook image seems to be enough. I merge this PR. |
Hi @alexnuttinck, I tested and it seems to work well in minikube. So in that respect I would approve the pull request. Note however that Tensorflow without GPU support has a limited interest; and I'm not sure that the tensorflow version installed in this docker image supports GPU. To be sure, it should be tested on a machine/cluster with a NVIDIA gpu and CUDA enabled. (Note that it is a more general remark than for tensorflow: other famous machine learning libraries are also far more efficient on GPUs, especially decision-trees-based libraries such as XGBoost or CatBoost). |
Hi @alexnuttinck @AyadiAmen @banzo , After investigation, I notice that from Tensorflow 2.0, the same python library (installed by pip) is used either for cpu and gpu usage: However, GPU support must still be enabled by installing the nvidia driver and CUDA. Infos about that here also: Also note that KubeFlow could also be an ultimate solution if it can be coupled/integrated with FADI (it already integrates various data science tools, including jupyter, ML workflow management and deployment): |
What this PR does / why we need it:
This PR updates the jupyterhub helm chart and adds a tensorflow notebook.
Special notes for your reviewer:
TensorBoard ( a tool for providing the measurements and visualizations needed during the machine learning workflow ) might be added later on using a separate helm chart.
Checklist
[Place an '[x]' (no spaces) in all applicable fields. Please remove unrelated fields.]