提交 713ed61f authored 作者: Nan Rosemary Ke's avatar Nan Rosemary Ke

add better tips

上级 0d04c13f
...@@ -702,10 +702,12 @@ Test them first, as they are not guaranteed to always provide a speedup.""") ...@@ -702,10 +702,12 @@ Test them first, as they are not guaranteed to always provide a speedup.""")
node = a[1] node = a[1]
if (isinstance(node.op, pool.Pool)): if (isinstance(node.op, pool.Pool)):
if not cuda.dnn.dnn_available(): if not cuda.dnn.dnn_available():
print('Need CuDNN for pooling') print("Install CuDNN to do pooling faster"
"this allows the operation to run on GPU")
if (isinstance(node.op, LogSoftmax)): if (isinstance(node.op, LogSoftmax)):
if not cuda.dnn.dnn_available(): if not cuda.dnn.dnn_available():
print('Need CuDNN for LogSoftmax') print("Install CuDNN to do LogSoftmax faster"
"this allows the operation to run on GPU")
if not printed_tip: if not printed_tip:
print(" Sorry, no tip for today.") print(" Sorry, no tip for today.")
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论