From ff2a8b32b1e6e6b833647c2e0af01b94dbc656fe Mon Sep 17 00:00:00 2001
From: Craig Warren <cwarren@C17776592.local>
Date: Fri, 8 Mar 2019 16:58:49 +0000
Subject: [PATCH] Updated lookup of CUDA devices.

---
 gprMax/utilities.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/gprMax/utilities.py b/gprMax/utilities.py
index ab55c00c..1766265d 100644
--- a/gprMax/utilities.py
+++ b/gprMax/utilities.py
@@ -388,6 +388,7 @@ def detect_check_gpus(deviceIDs):
 
     # Get list of available GPU device IDs
     deviceIDsavail = os.environ.get('CUDA_VISIBLE_DEVICES')
+    deviceIDsavail = [int(s) for s in deviceIDsavail.split(',')]
 
     # Print information about all detected GPUs
     gpus = []