From 872e97af847d24e447f335f956fcd8dcbb4a6931 Mon Sep 17 00:00:00 2001 From: Zhenwen Dai Date: Thu, 3 Jul 2014 16:31:22 +0100 Subject: [PATCH] fix the problem starting multiple process with limited number of GPUs --- GPy/util/gpu_init.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/GPy/util/gpu_init.py b/GPy/util/gpu_init.py index 845d38a1..b6a4a164 100644 --- a/GPy/util/gpu_init.py +++ b/GPy/util/gpu_init.py @@ -24,7 +24,7 @@ try: pycuda.driver.init() if gpuid>=pycuda.driver.Device.count(): print '['+MPI.Get_processor_name()+'] more processes than the GPU numbers!' - MPI.COMM_WORLD.Abort() + #MPI.COMM_WORLD.Abort() raise gpu_device = pycuda.driver.Device(gpuid) gpu_context = gpu_device.make_context()