4 ms·
AMD Graphics card support please?
by infocollector 8y ago
AMD Graphics card support please?
- dguaraglia 8y agoThere was a OpenCL branch no long ago that was supposed to make it into mainline at some point. You can (kinda) track progress of that here: https://github.com/tensorflow/tensorflow/issues/22 https://github.com/tensorflow/tensorflow/issues/22
- jlebar 8y agoWe are (well, AMD is) working on it. It's all being done in the open; there are PRs you can follow. You can even help out if you're so inclined. It's not going to be simple or fast, but it's coming... (I work on XLA, a compiler for TensorFlow, and I've been working closely with AMD on the TF/XLA -> AMDGPU port. My team also works on CUDA support in clang, and we've been reviewing AMD's patches to support HIP in upstream clang.)
- jonbaer 8y agohttps://github.com/ROCmSoftwarePlatform/tensorflow-upstream https://github.com/ROCmSoftwarePlatform/tensorflow-upstream