Gpu tensorflow example

doc/source/examples/gpu_tensorflow_deep_mnist.nblink

@@ -0,0 +1,3 @@
+{
+  "path": "../../../examples/models/gpu_tensorflow_deep_mnist/gpu_tensorflow_deep_mnist.ipynb"
+}

doc/source/examples/notebooks.rst

@@ -15,6 +15,7 @@ Notebooks
    Custom Endpoints <custom_endpoints>
    Example Helm Deployments <helm_examples>
    Go Model <go_example>
+   GPU Tensorflow Deep MNIST <gpu_tensorflow_deep_mnist>
    H2O Java MoJo <h2o_mojo>
    Istio Canary <istio_canary>
    Istio Examples <istio_examples>   

examples/models/gpu_tensorflow_deep_mnist/.s2i/environment

@@ -0,0 +1,4 @@
+MODEL_NAME=DeepMnist
+API_TYPE=REST
+SERVICE_TYPE=MODEL
+PERSISTENCE=0

examples/models/gpu_tensorflow_deep_mnist/DeepMnist.py

@@ -0,0 +1,17 @@
+import tensorflow as tf
+import numpy as np
+
+class DeepMnist(object):
+    def __init__(self):
+        self.class_names = ["class:{}".format(str(i)) for i in range(10)]
+        self.sess = tf.Session()
+        saver = tf.train.import_meta_graph("model/deep_mnist_model.meta")
+        saver.restore(self.sess,tf.train.latest_checkpoint("./model/"))
+
+        graph = tf.get_default_graph()
+        self.x = graph.get_tensor_by_name("x:0")
+        self.y = graph.get_tensor_by_name("y:0")
+
+    def predict(self,X,feature_names):
+        predictions = self.sess.run(self.y,feed_dict={self.x:X})
+        return predictions.astype(np.float64)

examples/models/gpu_tensorflow_deep_mnist/Makefile

@@ -0,0 +1,20 @@
+IMAGE_NAME=seldonio/gpu-mnist-example
+IMAGE_VERSION=0.1
+
+
+build_image:
+	s2i build . joelh1996/gpu-base:0.4 ${IMAGE_NAME}:${IMAGE_VERSION}
+
+push_image:
+	docker push $(IMAGE_NAME):$(IMAGE_VERSION)
+
+
+.PHONY: train
+train:
+	mkdir -p model
+	sudo python3 create_model.py
+
+.PHONY: clean
+clean:
+	rm -rf model
+	rm -rf MNIST_data