ray_lightning icon indicating copy to clipboard operation
ray_lightning copied to clipboard

Published 20 hours ago • verified publisher icon ray-project

Updates for Pytorch 2.0 and Pytorch-Lightning 2.0.1

Open noachr opened this issue 1 year ago • 0 comments

Quick changes to get RayStrategy working with latest pytorch and pytorch-lightning. Followed this guide primarily https://lightning.ai/docs/pytorch/stable/upgrade/from_1_6.html

Did not change Horovid and DDPSharded as I am not as familiar with them and they now use separate libaries.

Submitting in case this is useful to anyone.

noachr avatar Apr 03 '23 20:04 noachr