提交 717a3ab4 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Use warning() instead of info() to display message of init_gpu_device

上级 5da99b9e
...@@ -213,7 +213,7 @@ if config.device.startswith('gpu'): ...@@ -213,7 +213,7 @@ if config.device.startswith('gpu'):
use(device=config.device, force=config.force_device) use(device=config.device, force=config.force_device)
elif config.init_gpu_device: elif config.init_gpu_device:
assert config.device=="cpu", "We can use the Theano flag init_gpu_device only when the Theano flag device=='cpu'" assert config.device=="cpu", "We can use the Theano flag init_gpu_device only when the Theano flag device=='cpu'"
info(("GPU device %s will be initialized, and used if a GPU is needed. " warning(("GPU device %s will be initialized, and used if a GPU is needed. "
"However, no computation, nor shared variables, will be implicitly " "However, no computation, nor shared variables, will be implicitly "
"moved to that device. If you want that behavior, use the 'device' " "moved to that device. If you want that behavior, use the 'device' "
"flag instead.") % config.init_gpu_device) "flag instead.") % config.init_gpu_device)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论