search

ray-project / ray

Ray is a unified framework for scaling AI and Python applications. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.
https://ray.io
Apache License 2.0
33.11k stars 5.6k forks source link

Upgrade `pytorch_lightning` to `lightning` in all CI/release tests/examples #38200

Open woshiyyya opened 1 year ago

woshiyyya commented 1 year ago

Description

With version>2.0, the import path of PyTorch Lightning has been updated from import pytorch_lightning to import lightning.pytorch.

Although import pytorch_lightning still exists for backward compatibility, we have to modify our docker environment and lightning integration utilities to use the new import path.

Use case

No response

woshiyyya commented 1 year ago

cc @matthewdeng

ATheCoder commented 1 year ago

@woshiyyya @matthewdeng Is this on the road-map? I can do this as a first contribution if you guys are okay with it.