tensorflow中tf.ConfigProto()用法解释

本文转载自查看原文 2019-01-11 10:37 1383

在看C3D代码的时候，看见有一段代码是

config = tf.ConfigProto()#主要是配置tf.Session的运行方式，GPU还是CPU，在这里选择的是GPU的运行方式
config.gpu_options.allow_growth = True

 tf.ConfigProto()的主要用法就是配置tf.Session的运行方式，后一句指定的其为GPU的运行方式

本站转载的文章为个人学习借鉴使用，本站对版权不负任何法律责任。如果侵犯了您的隐私权益，请联系本站邮箱yoyou2525@163.com删除。

猜您在找 Tensorflow中tf.ConfigProto()详解 Tensorflow中tf.ConfigProto()详解 TensorFlow中tf.ConfigProto()配置Sesion运算方式 tf.ConfigProto() tensorflow中使用tf.ConfigProto()配置Session运行参数&&GPU设备指定 tf.Session()函数的参数应用（tensorflow中使用tf.ConfigProto()配置Session运行参数&&GPU设备指定）对 tensorflow 中 tf.nn.embedding_lookup 函数的解释 tensorflow中tf.keras.models.Sequential()用法【TensorFlow】tf.concat的用法 tensorflow中的tf.Variable()