张量流会话到底有什么作用?

dee*_*ing 5 python gpu machine-learning tensorflow

我安装了tensorflow的gpu版本,一旦我创建会话,它就会向我显示以下日志:

我tensorflow/core/common_runtime/gpu/gpu_device.cc:885]找到设备0,其属性:名称:GeForce GTX TITAN Black主要:3次要:5内存时钟速率(GHz)0.98 pciBusID 0000:01:00.0总内存:5.94GiB免费内存:5.31GiB 我tensorflow/core/common_runtime/gpu/gpu_device.cc:906] DMA:0我tensorflow/core/common_runtime/gpu/gpu_device.cc:916] 0:YItensorflow/core/common_runtime/gpu/gpu_device。 cc:975] 创建 TensorFlow 设备 (/gpu:0) ->(设备:0,名称:GeForce GTX TITAN Black,pci 总线 id:0000:01:00.0)

当我检查 GPU 内存使用情况时,大约 90% 都被消耗了。

Tensorflow 文档对此没有任何说明。它能控制GPU吗?为什么它会消耗大部分内存?

Ale*_*sos 2

TensorFlow 会话在启动时分配〜所有 GPU 内存,因此它们可以绕过 cuda 分配器。

不要在同一进程中运行多个使用 cuda 的库,否则会发生奇怪的事情(例如此流执行器错误)。