diff --git a/modules/util/setup.py b/modules/util/setup.py index 5b1f173a443525614950d73cc1d2c65dd5888506..32a67b229d082aafadd2d97d630f28c054117d49 100644 --- a/modules/util/setup.py +++ b/modules/util/setup.py @@ -1,7 +1,7 @@ # To support both python 2 and python 3 from __future__ import division, print_function, unicode_literals -import os +import os, sys from pathlib import Path @@ -29,12 +29,20 @@ cachepath = cachepath+'.pkl' try: user_paths = os.environ['PYTHONPATH'].split(os.pathsep) - python_path = user_paths[0] + os.sep + python_path = None + if len(user_paths) > 1: + for up in user_paths: + if up.find('cimss_icing') > 0: + python_path = up + os.sep + else: + python_path = user_paths[0] + os.sep + if python_path is None: + print('** No PYTHONPATH found containing cimss_icing/model *', file=sys.stderr) ancillary_path = python_path + 'ancillary' + os.sep model_path_day = python_path + 'tf_model_day' + os.sep model_path_night = python_path + 'tf_model_night' + os.sep except KeyError: - print('** PYTHONPATH must be defined! *') + print('** PYTHONPATH must be defined! *', file=sys.stderr) python_path = None gpu_devices = {0: '/device:GPU:0', 1: '/device:GPU:1', 2: '/device:GPU:2'}