Как GPU выполняет вычисления в Keras AAN? - PullRequest
Новая
       16

Как GPU выполняет вычисления в Keras AAN?

0 голосов
/ 08 октября 2019

Я знаю, что, возможно, этот пост повторяется, но моя проблема связана с Tensorflow 2.0.0. Я установил Tensorflow-GPU 2.0.0, но не вижу свой графический процессор при выполнении первой части кода ниже. Когда я использую Keras для AAN (вторая часть кода) и задаю параметр «n_jobs = -1» в переменной точности, тогда эпохи работают очень быстро, в отличие от метода по умолчанию для процессора. Графический процессор делает вычисления? 

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())

def build_classifier():
    classifier = Sequential()
    classifier.add(Dense(units = 6, kernel_initializer = 'uniform',activation = 'relu', input_dim = 11))
    classifier.add(Dense(units = 6, kernel_initializer = 'uniform', activation = 'relu'))
    classifier.add(Dense(units = 1, kernel_initializer = 'uniform', activation = 'sigmoid'))
    classifier.compile(optimizer = 'adam', loss = 'binary_crossentropy', metrics = ['accuracy'])
    return classifier
classifier = KerasClassifier(build_fn = build_classifier, batch_size = 10, epochs = 100)
accuracies = cross_val_score(estimator = classifier, X = X_train, y = y_train, cv = 10, n_jobs= -1)
...