tensorflow 進程完成，退出代碼為 -1073740791 (0xC0000409) STATUS_STACK_BUFFER_OVERRUN

Question

我正在嘗試按照本教程進行操作，但無法訓練第一個網絡。 它與此消息一起崩潰：

進程以退出代碼 -1073740791 (0xC0000409) 結束

並且不會拋出錯誤，所以我不知道該谷歌什么了

我正在使用 Pycharm 社區版和 python 3.7，我不是很有經驗。

我還搜索了不同的教程，但它們都非常相似。 我已經完成了構建 model 的所有工作，但培訓不會開始。 The tutorial used the model.fit_generate function but that threw an error so I am using the model.fit function. 我使用的fit function 錯了嗎？

這是我的代碼：

# baseline model for the dataset
import sys
from matplotlib import pyplot
from keras.utils import to_categorical
from keras.models import Sequential
from keras.layers import Conv2D
from keras.layers import MaxPool2D
from keras.layers import Dense
from keras.layers import Flatten
from keras.optimizers import SGD
from keras.preprocessing.image import ImageDataGenerator

# define cnn model
def define_model():
    model = Sequential()
    model.add(Conv2D(32, (3, 3), activation='relu', kernel_initializer='he_uniform', padding='same',
                     input_shape=(200, 200, 3)))
    model.add(MaxPool2D((2, 2)))
    model.add(Flatten())
    model.add(Dense(128, activation='relu', kernel_initializer='he_uniform'))
    model.add(Dense(1, activation='sigmoid'))
    # compile model
    opt = SGD(lr=0.001, momentum=0.9)
    model.compile(optimizer='adam', loss='binary_crossentropy', metrics=['accuracy'])
    model.summary()
    return model


# run the test harness for evaluating a model
def run_test_harness():
    # define model
    model = define_model()
    # create data generator
    datagen = ImageDataGenerator(rescale=1.0 / 255.0)
    # prepare iterators
    train_it = datagen.flow_from_directory('data_dogs_vs_cats/train', class_mode='binary', batch_size=64,
                                           target_size=(200, 200))
    test_it = datagen.flow_from_directory('data_dogs_vs_cats/val', class_mode='binary', batch_size=64,
                                          target_size=(200, 200))
    # fit model
    history = model.fit(train_it, steps_per_epoch=len(test_it), validation_data=test_it, validation_steps=len(test_it),
                        epochs=20, verbose=0)
    # evaluate model
    _, acc = model.evaluate_generator(test_it, steps=len(test_it), verbose=0)
    print('> %.3f' % (acc * 100.0))


# entry point
run_test_harness()

Answer 1

我遇到了這個問題，原因是 PyCharm 在直接從其中運行代碼時隱藏了錯誤。 嘗試從命令行/終端運行代碼，您將看到導致此崩潰的錯誤。

當 cuda 的 bin 目錄中缺少 DLL 或代碼中存在其他錯誤時，可能會發生這種情況。

在我的情況下，我的 cuda bin 文件夾（C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA）中缺少以下 DLL：

• cudnn64_8.dll
• cudnn_ops_train64_8.dll
• cudnn_ops_infer64_8.dll
• cudnn_cnn_infer64_8.dll

希望這可以幫助！