Transformer Lab Now Works with AMD GPUs

Transformer Lab Now Works with AMD GPUs(old.reddit.com)

2 points by aliasaria 356 days ago | 8 comments

aliasaria 356 days ago |

Getting ROCm working was... an adventure. We documented the entire (painful) journey in a detailed blog post because honestly, nothing went according to plan. If you've ever wrestled with ROCm setup for ML, you'll probably relate to our struggles.

The good news? Everything works smoothly now! We'd love for you to try it out and see what you think.

latchkey 356 days ago | |

Reading your post now, half the article feels like it is just installing PyTorch. Next time, just use the pre-built docker containers. It is the recommended way and much easier.

https://rocm.docs.amd.com/projects/install-on-linux/en/lates...

Additionally, our MI300x VMs/machines come with ROCm installed and configured already. We also apply all the default recommended BIOS settings as well.

anonymous_llama 356 days ago | | |

Hey, I think the major issue was getting torch 2.7 working on a bare-metal installation as the app needed that and the pre-built docker containers aren't out yet for torch 2.7. Transformer Lab is also much more than just the Pytorch setup as it provides several plugins which can be used for training, evaluation, dataset generation (and much more!).

Also, the pre-built docker containers unfortunately do not work on WSL which caused the majority of the issues.

I'd love to hear if you had a different experience or if I'm mistaken in any of this!

latchkey 356 days ago | |

No need to build your own box, we've got 1xMI300x VMs, for FREE (thanks to AMD), for development exactly like this. Reach out and we can get you set up.

Someone left a comment accusing me of advertising my business, then deleted it. If that’s how it came across, I apologize, but my intention was to offer something genuinely useful, for free, and directly relevant to helping the OP. Those credits weren’t easy to get, it required going all the way up to Lisa. I’m committed to making supercompute accessible to developers. Yes, it’s free like GitHub is free. But this isn’t a sales pitch.

anonymous_llama 356 days ago | | |

Thanks for offering the GPU and we'll definitely be in touch about this one. What is the best way to reach out to you?

latchkey 356 days ago |

Link to the source: https://transformerlab.ai/blog/amd-support/