From 5bc9c23c84a3e7f079b75d0bee50c570a7372920 Mon Sep 17 00:00:00 2001
From: Willem Jan Palenstijn <Willem.Jan.Palenstijn@cwi.nl>
Date: Wed, 6 Jan 2016 16:49:56 +0100
Subject: Allow multiple GPUs in ASTRA_GPU_INDEX envvar

---
 python/astra/__init__.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/python/astra/__init__.py b/python/astra/__init__.py
index 10ed74d..515d9a2 100644
--- a/python/astra/__init__.py
+++ b/python/astra/__init__.py
@@ -39,7 +39,7 @@ from . import log
 from .optomo import OpTomo
 
 import os
-try:
-    astra.set_gpu_index(int(os.environ['ASTRA_GPU_INDEX']))
-except KeyError:
-    pass
+
+if 'ASTRA_GPU_INDEX' in os.environ:
+    L = [ int(x) for x in os.environ['ASTRA_GPU_INDEX'].split(',') ]
+    astra.set_gpu_index(L)
-- 
cgit v1.2.3