litgpt icon indicating copy to clipboard operation
litgpt copied to clipboard
verified publisher icon Lightning-AI

Can I train a model on 7900XT 4 cards?

Open win10ogod opened this issue 1 year ago • 1 comments

Can I train a model on 7900XT 4 cards is rocm?

win10ogod avatar Mar 31 '24 07:03 win10ogod

We haven't tested AMD support. But it should generally work, the code might need minor modifications. Feel free to try and report back

carmocca avatar Apr 02 '24 17:04 carmocca