提交 8c5fef8a authored 作者: vesis84's avatar vesis84

don't call 'cuda_ndarray.select_a_gpu()' when using pycuda gpu selection,

上级 3af33c63
......@@ -404,7 +404,11 @@ def use(device,
pycuda_init_dev = theano.misc.pycuda_init.pycuda_available
try:
if (device != 'gpu') and not pycuda_init_dev:
if pycuda_init_dev:
use.device_number = active_device_number()
# This is needed to initialize the cublas handle.
gpu_init(use.device_number, config.lib.cnmem)
elif(device != 'gpu'):
assert isinstance(device, int)
gpu_init(device, config.lib.cnmem)
use.device_number = device
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论