tensorflow 进程已完成,退出代码为 -1073740791 (0xC0000409) STATUS_STACK_BUFFER_OVERRUN [重复]

问题描述 投票:0回答:1

我正在尝试遵循本教程,但我无法训练第一个网络。它崩溃并显示以下消息:

Process finished with exit code -1073740791 (0xC0000409)

并且不会抛出错误,所以我不知道该再谷歌什么

我使用的是 Pycharm 社区版和 python 3.7。

我还用谷歌搜索了不同的教程,但它们都非常相似。我已经完成了构建模型的所有工作,但训练无法开始。本教程使用了 model.fit_generate 函数,但引发了错误,所以我使用了 model.fit 函数。我是否错误地使用了 fit 函数?

这是我的代码:

# 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()
python tensorflow pycharm conv-neural-network
1个回答
0
投票

我遇到了这个问题,原因是 PyCharm 直接从内部运行代码时隐藏了错误。尝试从命令行/终端运行代码,您将看到哪些错误导致了此崩溃。

当 cuda 的 bin 目录中缺少 DLL 或代码中存在其他错误时,可能会发生这种情况。

在我的情况下,我的 cuda bin 文件夹(C:\Program Files\NVIDIA GPU 计算工具包\CUDA)中缺少以下 DLL:

  • cudnn64_8.dll
  • cudnn_ops_train64_8.dll
  • cudnn_ops_infer64_8.dll
  • cudnn_cnn_infer64_8.dll
© www.soinside.com 2019 - 2024. All rights reserved.