提交 ed996e17 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Better status reporting when importing pygpu

上级 b9eaaf7e
...@@ -9,8 +9,10 @@ _logger = logging.getLogger(_logger_name) ...@@ -9,8 +9,10 @@ _logger = logging.getLogger(_logger_name)
_logger.setLevel(logging.WARNING) _logger.setLevel(logging.WARNING)
error = _logger.error error = _logger.error
info = _logger.info
try: try:
import pygpu
import pygpu.gpuarray import pygpu.gpuarray
except ImportError: except ImportError:
pygpu = None pygpu = None
...@@ -39,7 +41,7 @@ def init_dev(dev): ...@@ -39,7 +41,7 @@ def init_dev(dev):
globals.kind = 'opencl' globals.kind = 'opencl'
devspec = dev[7:] devspec = dev[7:]
plat, dev = devspec.split(':') plat, dev = devspec.split(':')
devnum = int(dev)|(int(plat)>>16) devnum = int(dev)|(int(plat)<<16)
else: else:
globals.kind = None globals.kind = None
if globals.kind: if globals.kind:
...@@ -55,6 +57,14 @@ if pygpu: ...@@ -55,6 +57,14 @@ if pygpu:
theano.compile.shared_constructor(gpuarray_shared_constructor) theano.compile.shared_constructor(gpuarray_shared_constructor)
elif config.gpuarray.init_device != '': elif config.gpuarray.init_device != '':
init_dev(config.gpuarray.init_device) init_dev(config.gpuarray.init_device)
else:
info("pygpu support not configured, disabling")
pygpu = None
except Exception: except Exception:
error("Could not initialize pygpu, support disabled", exc_info=True) error("Could not initialize pygpu, support disabled", exc_info=True)
pygpu = None pygpu = None
else:
if (config.gpuarray.init_device != '' or
config.device.startswith('opencl') or
config.device.startswith('cuda')):
error("pygpu was configured but could not be imported", exc_info=True)
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论