CUDA_ERROR_OUT_OF_MEMORY in tensorflow

后端 未结 6 2032
盖世英雄少女心
盖世英雄少女心 2020-12-04 19:43

When I started to train some neural network, it met the CUDA_ERROR_OUT_OF_MEMORY but the training could go on without error. Because I wanted to use gpu memory

6条回答
  •  没有蜡笔的小新
    2020-12-04 19:57

    In case it's still relevant for someone, I encountered this issue when trying to run Keras/Tensorflow for the second time, after a first run was aborted. It seems the GPU memory is still allocated, and therefore cannot be allocated again. It was solved by manually ending all python processes that use the GPU, or alternatively, closing the existing terminal and running again in a new terminal window.

提交回复
热议问题