Low GPU usage by Keras / Tensorflow?

前端 未结 5 985
萌比男神i
萌比男神i 2021-02-12 18:01

I\'m using keras with tensorflow backend on a computer with a nvidia Tesla K20c GPU. (CUDA 8)

I\'m tranining a relatively simple Convolutional Neural Network, during tra

5条回答
  •  后悔当初
    2021-02-12 18:31

    Low GPU utilization might be due to the small batch size. Keras has a habit of occupying the whole memory size whether, for example, you use batch size x or batch size 2x. Try using a bigger batch size if possible and see if it changes.

提交回复
热议问题