tensorflow gpu 1.13.1

Author: root

examples/models/gpu_tensorflow_deep_mnist/DeepMnist.py

@@ -15,5 +15,3 @@ def __init__(self):
     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/Dockerfile

@@ -1,8 +1,13 @@
-FROM nvidia/cuda:9.0-cudnn7-runtime-ubuntu16.04
+FROM nvidia/cuda:10.0-cudnn7-runtime-ubuntu18.04
+# FROM nvidia/cuda:9.0-cudnn7-runtime-ubuntu16.04
 
 LABEL io.openshift.s2i.scripts-url="image:///s2i/bin"
 
 RUN apt-get update -y
+
+# RUN apt-get install -y python
+# RUN apt-get install -y python3
+
 RUN apt-get install -y python-pip python-dev build-essential
 RUN apt-get install -y python3-pip python3-dev
 
@@ -12,6 +17,7 @@ WORKDIR /microservice
 COPY ./s2i/bin/ /s2i/bin
 
 # keep install of seldon-core after the COPY to force re-build of layer
-RUN pip install seldon-core
+RUN pip3 install tensorflow-gpu==1.13.1
+RUN pip3 install seldon-core
 
 EXPOSE 5000

examples/models/gpu_tensorflow_deep_mnist/MNIST_data/t10k-images-idx3-ubyte.gz

@@ -3,7 +3,7 @@ IMAGE_VERSION=0.1
 
 
 build_image:
-	s2i build . joelh1996/gpu-base:0.1 ${IMAGE_NAME}:${IMAGE_VERSION}
+	s2i build . joelh1996/gpu-base:0.4 ${IMAGE_NAME}:${IMAGE_VERSION}
 
 push_image:
 	docker push $(IMAGE_NAME):$(IMAGE_VERSION)
@@ -12,7 +12,7 @@ push_image:
 .PHONY: train
 train:
 	mkdir -p model
-	python create_model.py
+	sudo python3 create_model.py
 
 .PHONY: clean
 clean:

examples/models/gpu_tensorflow_deep_mnist/MNIST_data/t10k-labels-idx1-ubyte.gz

@@ -0,0 +1,2 @@
+model_checkpoint_path: "deep_mnist_model"
+all_model_checkpoint_paths: "deep_mnist_model"