提交 e37988e5 authored 作者: Frédéric Bastien's avatar Frédéric Bastien

Merge pull request #1837 from abergeron/gpuarray_print_dev

Print current device when initializing gpuarray.
import sys
import logging import logging
import theano import theano
...@@ -36,6 +37,8 @@ def init_dev(dev): ...@@ -36,6 +37,8 @@ def init_dev(dev):
context = pygpu.init(dev) context = pygpu.init(dev)
pygpu.set_default_context(context) pygpu.set_default_context(context)
pygpu_activated = True pygpu_activated = True
if config.print_active_device:
print >> sys.stderr, "Using device %s: %s" % (dev, context.devname)
if pygpu: if pygpu:
try: try:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论