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