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 2731d578f321b284dc48b8f3445b417642b552da..bfab58e2480b5ee81e2d3090c2affc37e4804414 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