From 30fcddc0f9d803e8314af87095e38ef6b75c77cd Mon Sep 17 00:00:00 2001
From: Tru Huynh <tru@pasteur.fr>
Date: Mon, 20 Mar 2017 16:10:11 +0100
Subject: [PATCH] test added

---
 keras-tf-1.0.1-gpu-nvidia-367.57.sh | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/keras-tf-1.0.1-gpu-nvidia-367.57.sh b/keras-tf-1.0.1-gpu-nvidia-367.57.sh
index 2731d57..bfab58e 100644
--- a/keras-tf-1.0.1-gpu-nvidia-367.57.sh
+++ b/keras-tf-1.0.1-gpu-nvidia-367.57.sh
@@ -9,3 +9,14 @@ echo ${IMG_FILE}
 \rm -f ${IMG_FILE}
 sudo singularity create --size 12000 ${IMG_FILE}
 sudo singularity bootstrap ${IMG_FILE} ${IMG_DEF} 2>&1 | tee ${IMG_LOG}
+cat<<EOF | singularity exec ${IMG_FILE} python
+# Creates a graph.
+import tensorflow as tf
+a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a')
+b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2], name='b')
+c = tf.matmul(a, b)
+# Creates a session with log_device_placement set to True.
+sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
+# Runs the op.
+print sess.run(c)
+EOF
-- 
GitLab