Does Kohya support 5090s via the normal install?

[DewLover]

Been having trouble with finding a lora trainer to start learning with. Just got the 5090 but I see alot of mentions about special pytorch installs, etc. Would this trainer work out of the gate or does it require tweaks to get the new GPUs working?

[kohya-ss]

Unfortunately the maintainer does not have any RTX50x0 GPUs so they are not officially supported yet.

[araleza]

I'm using a 5090 successfully with kohya's sd-scripts. You just have to make sure you've installed the latest pytorch, which is as easy as picking the 12.8 CUDA version when installing torch from this page:

https://pytorch.org/get-started/locally/

Edit: I've seen some people saying that xformers isn't supported, but I wasn't using that anyway. I just use --sdpa in my command line and that seems fine.