1
基础环境
如何在海光上使用lightning.trainer?
2025-07-16 21:54:03
46

文档:https://lightning.ai/docs/pytorch/stable/common/trainer.html

如何安装?如何使用?(比如accelerator选择什么?)

全部评论(1)
只看楼主
C
changhl
1楼

老师,您好:

您是否提问的是pytorch lightning框架?

该框架为python层代码,无需适配,pip install 即可。

之前测试过,基于内部的pytorch2.3.0基础镜像,pip install 安装PyTorch Lightning 2.4.0版本即可。
2025/7/17
没有更多内容了
--
楼主
Level.2
0 帖子
楼主热帖
暂无数据