执行keras CNN模型不断使我的计算机崩溃

Question

我正在尝试使用R进行深度学习。我使用各种在线博客尝试测试其代码并查看其实际工作方式。对于keras，我不确定为什么，但是每次运行模型函数时，它都会崩溃。 很抱歉，如果我没有提供足够的信息。我正在运行AMD GPU和CPU  示例代码部分

history <- model %>% fit_generator(
  train_generator,
  steps_per_epoch = 100,
  epochs = 100,
  validation_data = validation_generator,
  validation_steps = 50,
)
use_multiprocessing=False

也

hist <- model %>% fit_generator(
  # training data
  train_image_array_gen,

  # epochs
  steps_per_epoch = as.integer(train_samples / batch_size), 
  epochs = epochs, 

  # validation data
  validation_data = valid_image_array_gen,
  validation_steps = as.integer(valid_samples / batch_size),

  # print progress
  verbose = 2,
  callbacks = list(
    # save best model after every epoch
    callback_model_checkpoint("C:/Users/My Account/Desktop/fruits_checkpoints.h5", save_best_only = TRUE),

  )

Answer 1

看来Keras使用tensorflow-gpu时出现了问题。安装tensorflow CPU版本后，尝试运行模型。由于您使用的是AMD GPU，因此可能无法将tensorflow gpu版本与cudnn库一起使用。