Ресурс исчерпан: OOM при выделении тензора с формой [32,128,768] и типом float on / job: localhost / replica: 0 / task: 0 / device: GPU: 0 - PullRequest
Новая
       39

Ресурс исчерпан: OOM при выделении тензора с формой [32,128,768] и типом float on / job: localhost / replica: 0 / task: 0 / device: GPU: 0

0 голосов
/ 18 апреля 2020

Я получаю эту ошибку после нескольких эпох. Я попробовал несколько предложений, найденных в похожих вопросах, таких как

  1. уменьшить размер пакета обучения и теста до 1
  2. уменьшить размер данных
  3. use kill - 9 pid
  4. использует более одного графического процессора, установив os.environ['CUDA_VISIBLE_DEVICES'] = '0,2'
  5. , уменьшив количество выходных нейронов модели LSTM
  6. , добавив gpu_options = tf.GPUOptions(allow_growth=True) session = tf.InteractiveSession(config=tf.ConfigProto(gpu_options=gpu_options))
  7. Добавление del model после использования модели
  8. Добавление k.clear_session(). Я не уверен, что правильно использовал именно этот.

Ни один из них не работает.

У кого-нибудь есть другие предложения? Пожалуйста помоги. Форма тензора изменяется в разных прогонах и в разных сообщениях об ошибках, но сообщение об ошибке остается неизменным.

Я использую Python3 .7, tenorflow-gpu == 1,14, CuDNN = 7.6.5, CUDA = = 10,0.

...