我需要哪些设置来调整 GPU 速度来训练神经网络？

【问题标题】：Which settings do I need to tune GPU speed to train Neural Networks?我需要哪些设置来调整 GPU 速度来训练神经网络？
【发布时间】：2018-05-09 15:15:39
【问题描述】：

我正在使用 Nvidia 显卡，并希望确保我能充分利用它们的性能（以安全的方式，不超频）。为了充分利用它们的功能，我需要执行哪些设置？

到目前为止，我用于更快启动的唯一设置是：

nvidia-smi -pm 1

【问题讨论】：

【解决方案1】：

您可以调整要牺牲多少 GPU。

config = tf.ConfigProto() config.gpu_options.per_process_gpu_memory_fraction = 0.4 session = tf.Session(config=config, ...)

例如，在官方文档中的这个示例中，您的程序将使用 GPU 总内存的 40%。

更多详情可以查看documentation

【讨论】：