使用PyTorch Lightning遇__init__()意外关键字参数'gpus'错误求助

问题原因 #

PyTorch Lightning 2.0及以上版本弃用了gpus这个旧参数，改用accelerator和devices来统一指定硬件配置，所以传入gpus会触发参数错误。

修复方法 #

将原来的Trainer初始化代码替换为以下两种方式之一：

1. 自动适配硬件（推荐）

trainer = pl.Trainer(max_epochs=20, accelerator="auto", devices=AVAIL_GPUS)
trainer.fit(model, dm)

accelerator="auto"会自动检测可用硬件（GPU/CPU/TPU等），devices指定使用的设备数量。

2. 明确指定GPU
   如果你确定要使用GPU，可以直接指定：

trainer = pl.Trainer(max_epochs=20, accelerator="gpu", devices=AVAIL_GPUS)
trainer.fit(model, dm)

额外说明 #

如果不确定当前PyTorch Lightning版本，可以运行以下代码查看：

import pytorch_lightning as pl
print(pl.__version__)

如果版本低于2.0，也可以选择降级到旧版本来适配视频中的代码，但更推荐使用新版本的参数规范。

内容的提问来源于stack exchange，提问作者Nam Tran