Update run_ATLAS_img.sh

run_ATLAS_img.sh

 #!/bin/bash
 
-`which nvidia-smi` 
-nvidia-smi 
-
 export ATLAS_LOCAL_ROOT_BASE=/cvmfs/atlas.cern.ch/repo/ATLASLocalRootBase;
-source ${ATLAS_LOCAL_ROOT_BASE}/user/atlasLocalSetup.sh -c centos7; # ATLAS image
-# asetup Athena,master,latest
-
-# source /cvmfs/sft.cern.ch/lcg/releases/LCG_101/tensorflow/2.5.0/x86_64-centos7-gcc11-opt/tensorflow-env.sh
-
-#lsetup "lcgenv -p LCG_101 x86_64-centos7-gcc11-opt tensorflow"
-#lsetup "lcgenv -p LCG_101_ATLAS_4 x86_64-centos7-gcc11-opt cuda"
-# lsetup "lcgenv -p LCG_101_ATLAS_4 x86_64-centos7-gcc11-opt cudnn"
-lsetup "views LCG_102b_cuda x86_64-centos7-gcc8-opt" prmon
 
-echo $LD_LIBRARY_PATH
+####### shifter ############
+source ${ATLAS_LOCAL_ROOT_BASE}/user/atlasLocalSetup.sh -c centos7 --swtype shifter;
+if [ $(env | grep "SHIFTER_RUNTIME") == 'SHIFTER_RUNTIME=1' ]
+    then echo "Running in container"
+else
+    echo "Error not in container...Exit..."
+    # exit
+fi
 
-prmon -i 1 -- python3 testtf.py
-exeStat=$?
-# exit $exeStat