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

adapt gpu initialization multiple gpu cards

Browse source
This commit is contained in:
Zhenwen Dai 2014-06-26 12:17:45 +01:00
parent cf33808673
commit bdf478956d
2 changed files with 27 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

30
GPy/util/gpu_init.py
View file

@ -6,17 +6,37 @@ providing CUBLAS handle: cublas_handle
"""
try:
import pycuda.autoinit
from scikits.cuda import cublas
import scikits.cuda.linalg as culinalg
culinalg.init()
cublas_handle = cublas.cublasCreate()
initSuccess = True
except:
initSuccess = False
gpu_initialized = False
gpu_device = None
gpu_context = None
def initGPU(gpuid=None):
if gpu_initialized:
return
if gpuid==None:
return pycuda.tools.make_default_context()
try:
import pycuda.autoinit
gpu_initialized = True
except:
pass
else:
return pycuda.driver.Device(gpuid).make_context()
try:
import pycuda.driver
pycuda.driver.init()
if gpuid>=pycuda.driver.Device.count():
return
gpu_device = pycuda.driver.Device(gpuid)
gpu_context = gpu_device.make_context()
gpu_initialized = True
except:
pass
def closeGPU():
if gpu_context is not None:
gpu_context.detach()