问在java中部署pytorch模型
EN

Stack Overflow用户

提问于 2020-03-17 19:50:11

回答 2查看 2.1K关注 0票数 0

我已经训练并保存了一个pytorch模型，现在我想在windows os中的java (不是android)环境中使用它(因为我使用的是一些仅在java中可用的库)，可以吗？我在pytorch文档中找不到直接的答案，当点击java api文档时，链接就断开了。

发布于 2020-04-09 09:31:54

@Gilad，你可以用Deep Java Library (djl.ai)做到这一点。查看：https://github.com/awslabs/djl/tree/master/pytorch/pytorch-engine

票数 2

发布于 2021-11-25 06:08:43

如上所述，您可以使用Deep Java Library完成此操作

但是，请注意，要使其正常工作，您需要将pytorch模型转换为torchscript模型。

票数 0

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/60721831

复制

相似问题

问在java中部署pytorch模型EN