AMD 的显卡，比如 RX 580 能用于 tensorflow 之类的机器学习吗？看网上教程都是 N 卡。

[coolair]

[GeruzoniAnsasu]

TF 有 cuda 模块，用 cuda 实现的 cnn 什么的

用 A 卡做通用计算要用 opencl，算法组老哥的意见是“比 cuda 难用很多”


搜了一下，大概有 opencl 实现的轮子，但可能不太圆，也不好炼

[mgbaozi]

https://medium.com/tensorflow/amd-rocm-gpu-support-for-tensorflow-33c78cc6a6cf
应该是 AMD 官方支持的

[minami]

可以，官方支持的，有 tf 和 caffe
https://rocm.github.io/dl.html

[coderluan]

能用，但是不好用，所以不推荐，学习阶段自己何必自找麻烦。