fixed example

Author: zyndagj

docs/05.container_gpu.md

@@ -125,10 +125,16 @@ for this purpose.
 ```
 # Change to your $WORK directory
 cd $WORK
+
 #Get the software
 git clone https://github.com/tensorflow/models.git ~/models
+
 # Pull the image
 singularity pull docker://gzynda/tacc-maverick-ml:latest
+
+# Is the GPU detected?
+singularity exec --nv tacc-maverick-ml-latest.img python -c "import tensorflow as tf; tf.test.is_gpu_available()"
+
 # Run the code
 singularity exec --nv tacc-maverick-ml-latest.img python $HOME/models/tutorials/image/mnist/convolutional.py
 ```
@@ -143,7 +149,7 @@ You may be thinking "what about overlayfs??". The Linux kernel on Maverick does
 You can also prototype your application from inside your container with the `shell` command.
 
 ```
-singularity shell tacc-maverick-ml-latest.img
+singularity shell --nv tacc-maverick-ml-latest.img
 python --version
 exit
 python --version