Tensorflow Keras 1.10如何强制使用所有CPU

时间:2018-09-28 06:51:35

标签: python tensorflow keras

如何强制model.fit方法使用所有cpu。

Tensorflow版本> 1.10

    import tensorflow as tf 
    from tensorflow import keras

    model = keras.models.Sequential([
        keras.layers.Dense(100, activation=tf.nn.relu, input_shape=(10,)),
        keras.layers.Dropout(0.5),
        keras.layers.Dense(50, activation=tf.nn.relu),
        keras.layers.Dense(10, activation=tf.nn.softmax)
    ])

    model.fit(train_data, target, epochs=100)

任何帮助都是非常有意义的。

0 个答案:

没有答案