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

fix the SSGPLVM with MPI

Browse source
This commit is contained in:
Zhenwen Dai 2014-06-25 13:57:25 +01:00
parent 08ed72b2f2
commit cf33808673
7 changed files with 28 additions and 30 deletions
Download patch file Download diff file Expand all files Collapse all files

8
GPy/util/gpu_init.py
View file

@ -13,4 +13,10 @@ try:
cublas_handle = cublas.cublasCreate()
initSuccess = True
except:
initSuccess = False
initSuccess = False
def initGPU(gpuid=None):
if gpuid==None:
return pycuda.tools.make_default_context()
else:
return pycuda.driver.Device(gpuid).make_context()