rocm下可用的 tensorflow 1.x 版本

tensorflow-rocm==1.13.0 (from versions: 1.8.0, 1.10.0, 1.10.1, 1.11.0, 1.12.0, 1.13.1, 1.13.2, 1.13.3, 1.13.4, 1.14.0, 1.14.1, 1.14.2, 1.14.3, 1.14.4, 1.14.5, 1.14.6, 1.15.0rc1, 1.15.0rc2, 1.15.0, 1.15.1, 1.15.2, 2.0.0b1, 2.0.0b2, 2.0.0b3, 2.0.0, 2.0.1, 2.0.2, 2.0.3, 2.1.0rc0, 2.1.0rc1, 2.1.0)

 

 

安装方法：

创建conda 的env

 

conda create -n tf-rocm python=3.5

conda activate tf-rocm

pip install tensorflow-rocm==xxx.xxx.xx   

上面的xx.xx.xx 就是版本号，从最上面选择一个，例 如：

pip install tensorflow-rocm==1.15.2

小提示：

在Rocm 3.1下，目前只有1.15.2可以正常工作，其它无法测试通过。