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可以正常工作,其它无法测试通过。