For image/object training what would be the most ideal

I have got amd graphics card and using Linux. I wanted to ask which platform would be the best and easiest to train models on my computer, using pytorch or something else such as yolo?