我对代码进行了转换,在nvidia-smi中,我可以看到它正在使用所有gpu的所有内存,每个gpu都在使用大约40%的内存,但每批培训的估计时间几乎与1个gpu相同。有人能引导我,告诉我如何在多个gpus上进行适当的训练吗?我的代码:from tensorflow.keras.layersimport Dense
from tensorflow
我使用TensorFlow2.0和Keras以及Sequential()应用程序接口来创建一个简单的模型: model = keras.Sequential(DROPOUT_RATE), keras.layers.Dropout(DROPOUT_RATEsparse_categorical_crossentropy",