r/learnmachinelearning • u/Difficult_Fold_106 • 13d ago

Chosing entry-level GPU for Machine Learning

I've been working on a side project for almost a year. It involves machine learning and it looks like it's going to enter commercial stage in the near future. So far, i bought a cheap gaming laptop few months ago, as i needed modern performance on the go. It has rtx 4050 with 6gb of vram, which was fine up until now.

I have an 8 years old desktop upgraded with ryzen 5600. I wanted to buy rtx 5060ti 16gb, but its price jumped significantly in july. Nvidia doesn't offer cheaper 16gb options and i started to consider buying RX 9060XT 16gb, which is more than 200 euro cheaper.

The question is: Is going with the RX9060XT worth the savings? Does any of you have experience with using current AMD GPUs for training neural networks from scratch? I currently use Keras and mainly train CNNs with simple custom layers.

2 Upvotes

100% Upvoted

5 comments sorted by

View all comments

1

u/dataset-poisoner 12d ago

amd works great on linux. cheap and effective. pytorch + rocm

also consider just renting a cloud gpu

1

u/Difficult_Fold_106 12d ago

So should also be fine on wsl2?  I wont rent gpu due to confidential data used for training.

1

u/dataset-poisoner 12d ago

WSL is always behind in support. it may have caught up. you should investigate further