cancel
Showing results for 
Search instead for 
Did you mean: 

AI Discussions

yamsyamsyams
Journeyman III

Ryzen 9 5900x - Machine Learning Training

Hi,

 

I'm not sure where to post this and I'm not sure if anyone else experiences this - so please let me know what other information I need to provide etc.

 

I work as a software developer and we train ML models. I was using scikit-learn's MLP library fine in training models and also TensorFlow v2.10. Whenever I try to use a higher version of TF or PyTorch for example, it will train for a few seconds and completely crash my computer. I have a suspicion this has something to do with my CPU. I do notice my CPU goes to 100% usage once it starts to train with either of the libraries I mentioned, and then it will crash after a few seconds.

 

Any help would be appreciated.

0 Likes
0 Replies