this post was submitted on 22 Jul 2023

4 points (100.0% liked)

Machine Learning

490 readers

1 users here now

A community for posting things related to machine learning

Icon base by Lorc under CC BY 3.0 with modifications to add a gradient

founded 1 year ago

MODERATORS

4

Training using AMD GPUs? (self.machine_learning)

submitted 1 year ago by catch22 to c/machine_learning

3 comments fedilink hide all child comments

Everything seems to be very Cuda centric, has anyone tried training or other ml projects using AMD GPUs? How was it?

top 3 comments

sorted by: hot top controversial new old

[–] lzbz 2 points 1 year ago

I used ROCm on my Vega 64 for my bachelor thesis. Once I got it to run, it was a huge improvemnt over training on CPU in terms of time. Didn't run into any problems, after the setup. Support for ROCm was very hit andmiss last time I checked though, so ymmv.

[–] [email protected] 1 points 1 year ago (1 children)

I think AMD supports pytorch now.

[–] const 1 points 11 months ago

And the other way around, mostly