Add a first check for gpus available

Author: AdrianoDee

Configuration/PyReleaseValidation/python/MatrixRunner.py

@@ -1,4 +1,5 @@
 import os, sys, time
+import subprocess
 
 from collections import Counter
 
@@ -16,9 +17,23 @@ def __init__(self, wfIn=None, nThrMax=4, nThreads=1, gpu=False):
         self.threadList = []
         self.maxThreads = nThrMax
         self.nThreads = nThreads
-        self.gpu = gpu
-
-        if self.gpu:
+        self.gpus = ()
+
+        if gpu:
+            print("> Running with --gpu option. Checking the GPUs available.")
+            cuda = subprocess.check_output("cudaComputeCapabilities", shell=True, executable="/bin/bash").decode('utf8')
+            # Building on top of the {cuda|rocm}ComputeCapabilities 
+            # output in case of no {NVIDIA|AMD} GPU:
+            # 'no XXX-capable device is detecte'
+            if "capable device is detected" in cuda:
+                cuda = 0
+            else:   
+                print(cuda.split("\n"))
+            rocm = subprocess.check_output("rocmComputeCapabilities", shell=True, executable="/bin/bash").decode('utf8')
+            if "capable device is detected" in rocm:
+                rocm = 0
+            else:
+                print(cuda.split("\n"))
             print("Checks for GPU")
             pass