2.12.0 之后不支持直接调用 GPU, 需要安装插件
pip install tensorflow-directml-plugin
安装过程中如果中途报 ERROR 也没关系, 最后一行显示 successfully 就行了
2023-04-16