如何提升GPU利用率 在使用fit_generator时

核心思想就是要减少读写次数,尽量提前尽可能的load数据,减少存储空间的交互,减少聚合过程

以下文字来源于Low GPU usage by Keras / Tensorflow? - Stack Overflowicon-default.png?t=M4ADhttps://stackoverflow.com/questions/44563418/low-gpu-usage-by-keras-tensorflow

The most possible scenarios are:

  • If you have a huge dataset, take a look at the disk read/write rates; if you are accessing your hard-disk frequently, most probably you need to change they way you are dealing with the dataset to reduce number of disk access

  • Use the memory to pre-load everything as much as possible.

  • If you are using a restful API or any similar services, make sure that you do not wait much for receiving what you need. For restful services, the number of requests per second might be limited (check your network usage via nmon/Task manager)

  • Make sure you do not use swap space in any case!

  • Reduce the overhead of preprocessing by any means (e.g. using cache, faster libraries, etc.)

  • Play with the bach_size (however, it is said that higher values (>512) for batch size might have negative effects on accuracy)

你可能感兴趣的:(keras,人工智能,深度学习)