AMD显卡GPU加速tensorflow的训练

安装tensorflow-directml即可，tensorflow-directml是tensorflow的一个分支，由微软在原版上修改得到的。使用方法与tensorflow是一致的。

官网链接：TensorFlow + DirectML 与 Windows ML | Microsoft Docs

安装命令：

pip install tensorflow-directml

注：建议在python虚拟环境中安装

注意事项：

①仅支持Windows系统，系统版本要求详见官方文档。Windows10系统可直接搜索运行dxdiag查看系统的版本号、DirectX版本等系统信息。

②对python版本的要求：现目前仅支持python3.5~3.7版本，具体版本支持的版本详见官方文档，或者去pypi官网，搜索tensorflow-directml，查看支持的python版本。

pypi官网：PyPI · The Python Package Index

③对显卡驱动的要求：确保显卡驱动为新版本，可通过系统更新进行更新。或者前往AMD官网下载最新显卡驱动。AMD 驱动程序和支持 | AMD

直接使用AMD官网提供的自动检测下载工具即可安装最新的AMD显卡驱动。(推荐使用Windows的自动更新，如果采用AMD官网的显卡驱动，后续Windows自动更新后又会装一个阉割版的AMD显卡驱动，都是显卡驱动的控制面板打不开！！！)

④安装tensorflow-directml后，在python中导入tensorflow时，如果出现如下报错与建议：

TypeError: Descriptors cannot not be created directly.

If this call came from a _pb2.py file, your generated code is out of date and must be regenerated with protoc >= 3.19.0.

If you cannot immediately regenerate your protos, some other possible workarounds are:

 1. Downgrade the protobuf package to 3.20.x or lower.

 2. Set PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION=python (but this will use pure-Python parsing and will be much slower).

可执行如下操作对 protobuf 进行降级：

pip uninstall protobuf

pip install protobuf==3.20.1

⑤使用keras下载的数据集位于：C:\Users\<用户名>\.keras 目录下

领个红包    支付宝     微信    support a 🍕