python - 有没有办法从 PyTorch 中删除未使用的文件以运行它的轻量版？

Question

您好我正在尝试上传所需库的 zip 存档以在 AWS Lambda 上运行我的项目。由于压缩的 PyTorch 库的大小超过了 AWS Lambda 的大小限制，我希望减少从库上传的文件数量。

我有一个训练有素的神经网络，我需要 PyTorch 来对 Lambda 进行推理。

是否有可以从包中删除的文件列表？有没有办法识别这些文件？

提前致谢 ：）

标签： pythonamazon-web-servicesaws-lambdaneural-networkpytorch

Answer 1

四处走走将使用 Lambda 对 EFS 的支持。您可以将您的依赖项预留到 EFS 中的文件夹结构，然后将 lambda 指向那里运行。然后，您应该能够导入 pytorch 和您放置为工作目录下的子目录的任何其他包。