提交 f4e48a80 authored 作者: notoraptor's avatar notoraptor

Code modified to let Theano print

PCI Bus ID ( see issue #5176: https://github.com/Theano/Theano/issues/5176 ). Tested both in Python 2 and Python 3. Works currently with CUDA, not with OpenCL ( see libgpuarray pull request #283: https://github.com/Theano/libgpuarray/pull/283 ).
上级 29af0e5b
......@@ -85,8 +85,8 @@ def init_dev(dev, name=None):
# This will map the context name to the real context object.
reg_context(name, context)
if config.print_active_device:
print("Mapped name %s to device %s: %s" %
(name, dev, context.devname),
print("Mapped name %s to device %s: %s; PCI Bus ID: %s" %
(name, dev, context.devname, context.pcibusid),
file=sys.stderr)
pygpu_activated = True
ctx_props = _get_props(name)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论