diff --git a/modules/util/setup.py b/modules/util/setup.py index e47599ded6129177c50ea5c316ab58e4afafd3aa..55ac45104c5b95cbf882242cc45ea564cca86876 100644 --- a/modules/util/setup.py +++ b/modules/util/setup.py @@ -27,6 +27,8 @@ root_cachedir = home_dir+'/cache' cachepath = "{}/run-{}".format(root_cachedir, now) cachepath = cachepath+'.pkl' +gpu_devices = {0: '/device:GPU:0', 1: '/device:GPU:1', 2: '/device:GPU:2'} + try: user_paths = os.environ['PYTHONPATH'].split(os.pathsep) python_path = None @@ -46,4 +48,3 @@ except KeyError: 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'}