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

also check for new backend

上级 fbaf1935
...@@ -701,11 +701,11 @@ Test them first, as they are not guaranteed to always provide a speedup.""") ...@@ -701,11 +701,11 @@ Test them first, as they are not guaranteed to always provide a speedup.""")
for a, t in iteritems(apply_time): for a, t in iteritems(apply_time):
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() or not theano.gpuarray.dnn.dnn_present():
print("Install CuDNN to do pooling faster" print("Install CuDNN to do pooling faster"
"this allows the operation to run on GPU") "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() or not theano.gpuarray.dnn.dnn_present():
print("Install CuDNN to do LogSoftmax faster" print("Install CuDNN to do LogSoftmax faster"
"this allows the operation to run on GPU") "this allows the operation to run on GPU")
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论