tensorflow中使用tf.ConfigProto()配置Session运行参数&&GPU设备指定

最新推荐文章于 2021-12-24 16:23:37 发布

原创

最新推荐文章于 2021-12-24 16:23:37 发布 · 3.4k 阅读

12 ·

CC 4.0 BY-SA版权

本文介绍如何使用tf.ConfigProto配置TensorFlow session，包括设备分配、软放置、GPU资源控制等关键技术细节。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

tf.ConfigProto一般用在创建session的时候。用来对session进行参数配置

config = tf.ConfigProto(log_device_placement=True, allow_soft_placement=True)

""" 
参数说明：
log_device_placement: 
设置tf.ConfigProto()中参数log_device_placement = True ,可以获取到operations 和 Tensor 
被指派到哪个设备(几号CPU或几号GPU)上运行,会在终端打印出各项操作是在哪个设备上运行的。

allow_soft_placement：
在tf中，通过命令 "with tf.device('/cpu:0'):",允许手动设置操作运行的设备。如果手动设置的设备
不存在或者不可用，就会导致tf程序等待或异常，为了防止这种情况，可以设置tf.ConfigProto()中参数
allow_soft_placement=True，允许tf自动选择一个存在并且可用的设备来运行操作。
"""