pytorch - Pytorch 闪电推理

Question

我使用多个 GPU 训练了一个模型，pytorch lightning并特别欣赏了使用多个 GPU 的便利性。现在在训练之后，我如何仍然利用闪电 GPU 功能在测试集上运行推理并存储/导出预测？

关于推理的文档并不针对此。

提前致谢。