Блог

Главная - Вопросы по программированию - Я могу найти свой графический процессор в tensorflow, но каждый раз, когда я запускаю, я отключаюсь от ядра

Я могу найти свой графический процессор в tensorflow, но каждый раз, когда я запускаю, я отключаюсь от ядра

#python #tensorflow #machine-learning #gpu #spyder

#питон #тензорный поток #машинное обучение #графический процессор #спайдер

Вопрос:

Я хочу запустить модель CNN, она называется VGG16 в tensorflow, я следовал правилам настройки GPU в tensorflow и создал новый env через anaconda, я также попытался переустановить все. Я не смог найти соответствующую ошибку в Google. Пожалуйста, помогите мне, пожалуйста

 train_y = np_utils.to_categorical(train_y) test_y = np_utils.to_categorical(test_y) train_X = train_X.astype("float32") test_X = train_X.astype("float32") vgg = VGG16(input_shape = (60,80,3), include_top=False) vgg.summary() type(vgg) model = Sequential() for layer in vgg.layers:  model.add(layer) model.layers.pop() for layer in model.layers:  layer.trainable = True model.add(Flatten()) model.add(Dense(5, activation= 'softmax'))  model.summary() model.compile(optimizer='adam', loss='categorical_crossentropy',  metrics=['accuracy'])  history = model.fit(train_X, train_y, epochs=15)  print(history)

все нормально

Комментарии:

1. Запустите код в терминале за пределами jupyter, и он, вероятно, выдаст вам лучшее сообщение об ошибке, которое вы можете включить в свой вопрос.

2. ( Сопровождающий Spyder здесь ) Какова ваша версия Spyder?

Метки: , , ,