tensorflow会话的具体功能是什么？

Question

我已经安装了tensorflow的gpu版本，只要我创建一个会话，它就会显示以下日志：

I tensorflow/core/common_runtime/gpu/gpu_device.cc:885] Found device 0 with properties: name: GeForce GTX TITAN Black major: 3 minor: 5 memoryClockRate (GHz) 0.98 pciBusID 0000:01:00.0 Total memory: 5.94GiB Free memory: 5.31GiB I tensorflow/core/common_runtime/gpu/gpu_device.cc:906] DMA: 0 I tensorflow/core/common_runtime/gpu/gpu_device.cc:916] 0: Y I tensorflow/core/common_runtime/gpu/gpu_device.cc:975] Creating TensorFlow device (/gpu:0) -> (device: 0, name: GeForce GTX TITAN Black, pci bus id: 0000:01:00.0)

当我检查我的GPU内存使用情况时，大约90%的内存被消耗掉了。在

Tensorflow文档没有对此做任何说明。它控制gpu吗？为什么它会消耗大部分内存？在

Answer 1

TensorFlow会话在启动时分配~所有GPU内存，因此可以绕过cuda分配器。在

不要在同一个进程中使用库运行多个cuda，否则会发生奇怪的事情（比如这个流执行器错误）。在