请参阅以下网址:
Installation — PyTorch Lightning 2.0.7 documentation
版本对应信息是:
Versioning Policy — PyTorch Lightning 2.0.7 documentation
简单安装方法:
使用conda create -n lgt python=3.11创建虚拟环境lgt后
使用conda activate lgt进入
直接使用python -m pip install lightning
这会自动install所匹配的torch包
如果先install torch,在install lightning,很大可能会报错