0

torch.jit.trace我有一个使用函数导出的 PyTorch 模型。我想将它容器化,并且我想了解使它工作的最低点子要求是什么。我的目标是获得一个纤薄的 Docker 镜像。具体来说,在我看来,我需要安装整个torch软件包。这个对吗?是否有其他方法来部署 TorchScript 模型/PyTorch 模型,这会导致更小的 Docker 映像(即可以避免安装整个torch包)?

4

0 回答 0