ROCm平臺搭建tensorflow2.0 踩坑日記

tensorflow-addon在我寫在這篇博文時還未被支持 RNN style的layer請使用tf.compat.v1.keras.layers.LSTM而不是tf.keras.layers.LSTM,由於後者默認開啓CuDNN加速,AMD顯卡顯然是不支持的
相關文章
相關標籤/搜索