apunkt/GPy
1
0
Fork 0
mirror of https://github.com/SheffieldML/GPy.git synced 2026-06-11 15:15:15 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix the problem starting multiple process with limited number of GPUs

Browse source
This commit is contained in:
Zhenwen Dai 2014-07-03 16:31:22 +01:00
parent a7d0438731
commit 872e97af84
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
GPy/util/gpu_init.py
View file

@ -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()