Hacker Newsnew | past | comments | ask | show | jobs | submitlogin

Tensorflow does officially support ROCm. The project was started by AMD and later upstreamed.

https://github.com/tensorflow/tensorflow/tree/master/tensorf...

https://github.com/tensorflow/tensorflow/blob/master/tensorf...

It is true that it is not Google who are distributing binaries compiled with ROCm support through PyPI (tensorflow and tensorflow-gpu is uploaded by Google, but tensorflow-rocm is uploaded by AMD). Is this what you meant by "not officially supporting"?



What you describe sounds a lot like the PyTorch support before this announcement: You could download PyTorch from AMD's ROCm site or build it yourself for >= 2 years now and this worked very reliably. (Edit: The two years (Nov 2018 or so) are the ones I can attest to from using it personally, but it probably didn't start then.)

The news here is that the PyTorch team and AMD are confident enough about the quality that they're putting it on the front page. This has been a long way in the making, and finally achieving official support is a great step for the team working on it.


Oh, interesting. I do wonder if Google puts the same quality control and testing into the Rocm version, though. Otherwise it would really be a lower tier of official support.

Granted, I don't know anything about the quality of PyTorch's support either.




Guidelines | FAQ | Lists | API | Security | Legal | Apply to YC | Contact

Search: