Fix to free resources after application is completed

cerr · May 6, 2019 · 50c79fc · 50c79fc
1 parent 920822b
commit 50c79fc
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 12 deletions.
diff --git a/CERR_core/Contouring/models/mr_prostate_DeepLab/MR_Prost_DeepLab_Recipe.txt b/CERR_core/Contouring/models/mr_prostate_DeepLab/MR_Prost_DeepLab_Recipe.txt
@@ -66,25 +66,18 @@ From: nvidia/cuda:9.0-cudnn7-devel-ubuntu16.04
 
   	# Install expect to automate responses for interactive build questions
   	apt-get -y install expect
-
-  	apt-get clean
-  	apt-get dist-upgrade && apt-get install -y --no-install-recommends --ignore-missing \
-  	python-dicom
-
-
+
   	apt update
   	apt-get clean
 
 	apt update && apt install -y libsm6 libxext6
   	apt-get install -y libxrender-dev
-
-  	pip3 install pydicom
+
   	pip3 install numpy
   	pip3 install scikit-image
   	pip3 install opencv-python
   	pip3 install protobuf
-  	pip3 install matplotlib
-
+
 	apt-get update 
 	apt-get install -y libgtk2.0-dev
 
@@ -106,7 +99,8 @@ From: nvidia/cuda:9.0-cudnn7-devel-ubuntu16.04
 
 	. /env/bin/activate	
 
-	python /software/CERR/CERR_core/Contouring/models/mr_prostate_DeepLab/run_inference_clinical_3D.py arg1=$1 arg2=$2	
+	python /software/CERR/CERR_core/Contouring/models/mr_prostate_DeepLab/run_inference_clinical_3D.py arg1=$1 arg2=$2
+	pkill -f /software/CERR/CERR_core/Contouring/models/mr_prostate_DeepLab/run_inference_clinical_3D.py	
 
 %test
 	#Test that script is a success

diff --git a/CERR_core/Contouring/models/mr_prostate_DeepLab/run_inference_clinical_3D.py b/CERR_core/Contouring/models/mr_prostate_DeepLab/run_inference_clinical_3D.py
@@ -259,6 +259,6 @@ def main(argv):
         with h5py.File(os.path.join(save_dir, maskfilename), 'w') as hf:
             hf.create_dataset("mask", data=mask)
         # os.remove(filename)
-
+        sys.exit()
 if __name__ == '__main__':
   tf.app.run()