如何使用Keras進行分佈式/多GPU運算?

如何使用Keras進行分佈式/多GPU運算? Keras在使用TensorFlow做爲後端的時候能夠進行分佈式/多GPU的運算,Keras對多GPU和分佈式的支持是經過TF完成的。html with tf.device('/gpu:0'): x = tf.placeholder(tf.float32, shape=(None, 20, 64)) y = LSTM(32)(x) #
相關文章
相關標籤/搜索