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