首页 > 解决方案 > 如何在 Windows 32 位机器上部署深度学习模型?

问题描述

我已经用 tensorflow 为 ocr 训练了一个 crnn 模型。然后我在 Windows 64 位机器上使用 tensorflow c++ API 部署它,它运行良好。现在我想在 windows 32 位机器上部署模型,但我注意到大多数深度学习框架,如 tensorflow 和 pytorch 不支持 32 位机器。然后我尝试了opencv dnn模块和ncnn框架,但它们都不支持我模型中使用的squeeze和unsqueeze操作。现在我不知道,有没有人可以给我一些建议?谢谢。

标签: tensorflowdeep-learningpytorch

解决方案


最接近您的问题的可能是这个 https://github.com/VisionRush/Caffe-Win32

这是 Windows 32 位上的 caffe。

P/S:这是一个仅 CPU 的构建。CpuToCpu对于 GPU 支持,您必须稍微调整一下 caffe 源代码,例如CpuToGpu. 但是,您将无法将 cuDNN 与 32 位 caffe 一起使用。


推荐阅读