Skip to content
This repository has been archived by the owner on Nov 3, 2023. It is now read-only.

Updates for Pytorch 2.0 and Pytorch-Lightning 2.0.1 #256

Open
wants to merge 4 commits into
base: main
Choose a base branch
from

Conversation

noachr
Copy link

@noachr noachr commented Apr 3, 2023

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.

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant