标签: theano keras cudnn
我在Keras(带有Theano后端)实现了9层卷积神经网络,并从另一个网络转换了预先训练的权重。但是,虽然在C中实现的原始网络中只需不到1秒,但是Theano对于单个图像需要10秒钟。
我在3 GB Quadro K4000上运行并安装了cuda-repo-rhel7-7.5和cudnn-7.0-linux-x64-v3.0。我做错了吗?
cuda-repo-rhel7-7.5
cudnn-7.0-linux-x64-v3.0