search

Azure / kubeflow-labs

👩‍🔬 Train and Serve TensorFlow Models at Scale with Kubernetes and Kubeflow on Azure
Creative Commons Attribution 4.0 International
289 stars 99 forks source link

Need example with details about wbuchwalter/tf-mnist:gpu #41

Closed cheyang closed 6 years ago

cheyang commented 6 years ago

In tfjob introduction , it mentions wbuchwalter/tf-mnist:gpu. Can you please provide details about how to build this docker image? Thanks.

ritazh commented 6 years ago

The source for the image is here: https://github.com/Azure/kubeflow-labs/blob/master/1-docker/src/

You can follow https://github.com/Azure/kubeflow-labs/tree/master/1-docker#building-the-image to build and push the image yourself.

cheyang commented 6 years ago

Thank you for reply. I need to confirm. Is it for GPU？ Looks like it's for docker image only with CPU support.