Update run_img.sh

run_img.sh

@@ -9,22 +9,20 @@ else
     # exit
 fi
 
-# export ATLAS_LOCAL_ROOT_BASE=/cvmfs/atlas.cern.ch/repo/ATLASLocalRootBase;
-# source ${ATLAS_LOCAL_ROOT_BASE}/user/atlasLocalSetup.sh;
-# # asetup Athena,master,latest
+local_sitepackage_path=`python -c "from distutils.sysconfig import get_python_lib; print(get_python_lib())"`
 
-# # source /cvmfs/sft.cern.ch/lcg/releases/LCG_101/tensorflow/2.5.0/x86_64-centos7-gcc11-opt/tensorflow-env.sh
+export ATLAS_LOCAL_ROOT_BASE=/cvmfs/atlas.cern.ch/repo/ATLASLocalRootBase;
+source ${ATLAS_LOCAL_ROOT_BASE}/user/atlasLocalSetup.sh;
+# asetup Athena,master,latest
 
-# #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
+# setting these after the ALRB setup so they are picked over ALRB packages
+PYTHONPATH=$local_sitepackage_path
+echo $PYTHONPATH
 
-# prmon -i 1 -- python3 testtf.py
-which python
-# source <path of the venv>/bin/activate
-python -c "import graph_nets"
+python -m pip list
+
+python -c "import graph_nets,numpy;print(graph_nets.print_function);print(numpy.__version__)"
 exeStat=$?
 # exit $exeStat