From 559e41800d576b95f09a277af624f3185e2a4f5b Mon Sep 17 00:00:00 2001 From: tomrink <rink@ssec.wisc.edu> Date: Fri, 2 Jun 2023 10:29:32 -0500 Subject: [PATCH] snapshot... --- modules/util/setup.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/modules/util/setup.py b/modules/util/setup.py index e47599de..55ac4510 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'} -- GitLab