TensorFlow Session 中关于 GPU 的配置项解析 ——转自 慢慢学TensorFlow 微信公众号
之前困扰笔者很久的一个 TensorFlow 问题就是显存占用太高,跑一个最简单的 MNIST 例程居然就能把整个 GTX 1080 显存用满!导致其他需要使用 GPU 的程序运行时出现 Out of Memory 的错误。本文将介绍如何在创建 TensorFlow Session 对象时进行 GP...
之前困扰笔者很久的一个 TensorFlow 问题就是显存占用太高,跑一个最简单的 MNIST 例程居然就能把整个 GTX 1080 显存用满!导致其他需要使用 GPU 的程序运行时出现 Out of Memory 的错误。本文将介绍如何在创建 TensorFlow Session 对象时进行 GP...