sess中设备指派gpu使用

最新推荐文章于 2021-01-26 02:32:45 发布

转载最新推荐文章于 2021-01-26 02:32:45 发布 · 431 阅读

文章标签：

#tensorflow

学习专栏收录该内容

12 篇文章

订阅专栏

转载https://blog.youkuaiyun.com/dcrmg/article/details/79091941

**
tf.ConfigProto()函数用在创建session的时候，用来对session进行参数配置
在这里插入图片描述

①

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

②自动选择运行设备

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