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

check for or

上级 fc6d12b3
...@@ -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() or not theano.gpuarray.dnn.dnn_present(): if not cuda.dnn.dnn_available() and 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() or not theano.gpuarray.dnn.dnn_present(): if not cuda.dnn.dnn_available() and 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 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论