4 ms·
ROCm is supported natively in TF 2.0, so they will benefit from that.
by useful 7y ago
ROCm is supported natively in TF 2.0, so they will benefit from that.
- PeterisP 7y agoIs it? TF2.0 Alpha and Beta release notes don't mention anything about it, and the official docs at https://www.tensorflow.org/install/pip https://www.tensorflow.org/install/pip are quite explicit that the tensorflow-gpu supports only CUDA cards. The last time I looked at ROCm (which was something like a year ago) it was "supported" as in "there's reports on the internet that someone got it to work" but when I tried, I couldn't get it to work and it really wasn't worth the effort. If I'd dig out a machine with an AMD GPU right now, can I get it working (like, train MNIST or some other helloworld'y system) within an hour, and is there documentation available on how exactly that should be done?
- imtringued 7y agohttps://pypi.org/project/tensorflow-rocm/ https://pypi.org/project/tensorflow-rocm/