tensorflow gpu 1.13.1

SeldonIO · Jun 17, 2019 · feb582e · feb582e
1 parent a7c4a19
commit feb582e
Show file tree

Hide file tree

Showing 13 changed files with 14 additions and 8 deletions.
diff --git a/examples/models/gpu_tensorflow_deep_mnist/DeepMnist.py b/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)
-
-
diff --git a/examples/models/gpu_tensorflow_deep_mnist/Dockerfile b/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
diff --git a/examples/models/gpu_tensorflow_deep_mnist/MNIST_data/t10k-images-idx3-ubyte.gz b/examples/models/gpu_tensorflow_deep_mnist/MNIST_data/t10k-images-idx3-ubyte.gz
diff --git a/examples/models/gpu_tensorflow_deep_mnist/MNIST_data/t10k-labels-idx1-ubyte.gz b/examples/models/gpu_tensorflow_deep_mnist/MNIST_data/t10k-labels-idx1-ubyte.gz
diff --git a/examples/models/gpu_tensorflow_deep_mnist/MNIST_data/train-images-idx3-ubyte.gz b/examples/models/gpu_tensorflow_deep_mnist/MNIST_data/train-images-idx3-ubyte.gz
diff --git a/examples/models/gpu_tensorflow_deep_mnist/MNIST_data/train-labels-idx1-ubyte.gz b/examples/models/gpu_tensorflow_deep_mnist/MNIST_data/train-labels-idx1-ubyte.gz
diff --git a/examples/models/gpu_tensorflow_deep_mnist/Makefile b/examples/models/gpu_tensorflow_deep_mnist/Makefile
@@ -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:

diff --git a/examples/models/gpu_tensorflow_deep_mnist/model/checkpoint b/examples/models/gpu_tensorflow_deep_mnist/model/checkpoint
@@ -0,0 +1,2 @@
+model_checkpoint_path: "deep_mnist_model"
+all_model_checkpoint_paths: "deep_mnist_model"
diff --git a/examples/models/gpu_tensorflow_deep_mnist/model/deep_mnist_model.data-00000-of-00001 b/examples/models/gpu_tensorflow_deep_mnist/model/deep_mnist_model.data-00000-of-00001
diff --git a/examples/models/gpu_tensorflow_deep_mnist/model/deep_mnist_model.index b/examples/models/gpu_tensorflow_deep_mnist/model/deep_mnist_model.index
diff --git a/examples/models/gpu_tensorflow_deep_mnist/model/deep_mnist_model.meta b/examples/models/gpu_tensorflow_deep_mnist/model/deep_mnist_model.meta
diff --git a/examples/models/gpu_tensorflow_deep_mnist/requirements.txt b/examples/models/gpu_tensorflow_deep_mnist/requirements.txt
@@ -1 +1 @@
-tensorflow-gpu==1.12.0
+tensorflow-gpu==1.13.1
diff --git a/examples/models/gpu_tensorflow_deep_mnist/s2i/bin/assemble b/examples/models/gpu_tensorflow_deep_mnist/s2i/bin/assemble
@@ -52,7 +52,7 @@ cp -Rf /tmp/src/. ./
 
 if [[ -f requirements.txt ]]; then
   echo "---> Installing dependencies ..."
-  pip install --find-links /whl -r requirements.txt
+  pip3 install --find-links /whl -r requirements.txt
 elif [[ -f setup.py ]]; then
   echo "---> Installing application ..."
   pip install --find-links /whl -e .
Original file line number	Diff line number	Diff line change
Expand Up		@@ -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)
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,2 @@
		model_checkpoint_path: "deep_mnist_model"
		all_model_checkpoint_paths: "deep_mnist_model"