为AMD RX 580安装TensorFlow遇版本匹配错误,求解决方法
针对RX580 GPU训练模型时TensorFlow安装错误的解决方案
问题根源
tensorflow-rocm仅支持Linux平台,Windows系统下无法通过pip获取该包,直接安装必然失败。- 旧版
tensorflow-directml包已停止维护,官方不再提供pip分发版本,因此会出现"找不到匹配版本"的错误。
解决步骤
1. 切换至Windows下AMD GPU的官方推荐方案
目前微软将DirectML整合为TensorFlow的插件,无需依赖旧版专属包,具体操作:
- 卸载残留的TensorFlow相关包:
pip uninstall tensorflow tensorflow-directml -y - 安装兼容的标准TensorFlow版本(推荐2.10.x至2.15.x,适配Python 3.9):
pip install tensorflow==2.15.0 - 安装DirectML插件以启用AMD GPU加速:
pip install tensorflow-directml-plugin - 验证GPU识别:运行以下代码确认RX580被检测到:
输出包含你的GPU设备信息即表示配置成功。import tensorflow as tf print(tf.config.list_physical_devices('GPU'))
2. 额外排查点
- 确保操作在激活的虚拟环境中执行(如果使用了虚拟环境)。
- 确认Windows版本为10 1903及以上或Windows 11,DirectML不支持更早的系统版本。
内容的提问来源于stack exchange,提问作者user21525821
相关产品推荐
相关产品推荐

