Allow pytorch 2.0 & pytorch-lightning 2.0

[alexandervaneck]

Hi, I'm just getting started with pycave and would like to use pytorch 2.0. It's torch.compile mechanism could be really powerful.

Would you please allow your package to also allow 2.0 for both pytorch and pytorch-lightning?

Also, would you expected any speedups from using torch.compile?

Thank you! 🙇

[farisjanjos]

Would also like to inquire if there are any plans to bump to pytorch-2.0 :) I'm planning to publish code for a recently submitted paper where we used PyCave (and cited it!). It would make our lives much easier if we could use torch 2.0 instead of fixing to torch1.12.1+cu116, which is what we had to use.

[borchero]

Would love to do that. Unfortunately, I haven't used the latest versions of PyTorch (Lightning) and torchmetrics so I will need some time to figure out relevant changes. If you want to provide some support fixing the CI for the latest versions, please go ahead 😅

[AndreasGerken]

Would love to use a higher version aswell.