在看C3D代码的时候,看见有一段代码是
config = tf.ConfigProto()#主要是配置tf.Session的运行方式,GPU还是CPU,在这里选择的是GPU的运行方式 config.gpu_options.allow_growth = True
tf.ConfigProto()的主要用法就是配置tf.Session的运行方式,后一句指定的其为GPU的运行方式
在看C3D代码的时候,看见有一段代码是
config = tf.ConfigProto()#主要是配置tf.Session的运行方式,GPU还是CPU,在这里选择的是GPU的运行方式 config.gpu_options.allow_growth = True
tf.ConfigProto()的主要用法就是配置tf.Session的运行方式,后一句指定的其为GPU的运行方式