提交 9f691454 authored 作者: Frederic Bastien's avatar Frederic Bastien

Print cudnn version when staring gpuarray back-end

上级 adb02ace
......@@ -64,7 +64,13 @@ def init_dev(dev, name=None):
reg_context(name, context)
pygpu_activated = True
if config.print_active_device:
print("Mapped name %s to device %s: %s" % (name, dev, context.devname),
cudnn_version = "not available"
try:
cudnn_version = dnn.version()
except Exception:
pass
print("Mapped name %s to device %s: %s (CuDNN version %s)" % (
name, dev, context.devname, cudnn_version),
file=sys.stderr)
# This maps things like 'cuda0' to the context object on that device.
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论