ROCm平台搭建tensorflow2.0 踩坑日记

tensorflow-addon在我写在这篇博文时还未被支持 RNN style的layer请使用tf.compat.v1.keras.layers.LSTM而不是tf.keras.layers.LSTM,由于后者默认开启CuDNN加速,AMD显卡显然是不支持的
相关文章
相关标签/搜索