this post was submitted on 22 Jul 2023
4 points (100.0% liked)

Machine Learning

472 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
[email protected]
[email protected]
[email protected]
4
Training using AMD GPUs? (programming.dev)
submitted 1 year ago by [email protected] to c/[email protected]
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
[–] [email protected] 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.

[–] [email protected] 1 points 9 months ago

And the other way around, mostly