提交 ed86748d authored 作者: Frederic Bastien's avatar Frederic Bastien

fix the init of the gpu device when their is none.

上级 d18089cb
......@@ -13,6 +13,19 @@ import theano.compile.sandbox
import logging, os
import logging, copy
_logger_name = 'theano_cuda_ndarray'
_logger = logging.getLogger(_logger_name)
_logger.setLevel(logging.WARN)
_logger.addHandler(logging.StreamHandler())
def warning(*msg):
_logger.warning(_logger_name+'WARNING: '+' '.join(str(m) for m in msg))
def info(*msg):
_logger.info(_logger_name+'INFO: '+' '.join(str(m) for m in msg))
def debug(*msg):
_logger.debug(_logger_name+'DEBUG: '+' '.join(str(m) for m in msg))
def use(device=None):
if use.device_number is None:
# No successful call to use() has been made yet
......@@ -27,7 +40,6 @@ def use(device=None):
use.device_number = device
except RuntimeError, e:
logging.getLogger('theano_cuda_ndarray').warning("WARNING: Won't use the GPU as the initialisation of device %i failed. %s" %(device, e))
raise
elif use.device_number != device:
logging.getLogger('theano_cuda_ndarray').warning("WARNING: ignoring call to use(%s), GPU number %i is already in use." %(str(device), use.device_number))
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论