美文网首页
TensorFlow的tips

TensorFlow的tips

作者: haoshengup | 来源:发表于2018-06-02 16:14 被阅读40次

节点和张量

节点张量铺设好了之后,只要不加sess.run(),可以运行脚本检查张量节点是否匹配,无需传入实际数据流。

'conv1'指节点,'conv1:0'指节点输出的第一个张量。

sess上下文环境中的函数调用即使不传入sess句柄,函数体内也存在于默认的sess环境中,可以直接sess.run()。

image_holder = tf.placeholder(tf.float32,[None, None, None, 3]),使用None回避指定大小。

GPU模式禁用

import os

os.environ["CUDA_VISIBLE_DEVICES"]="-1"

GPU资源申请规则

# 设置 GPU 按需增长

config = tf.ConfigProto()

config.gpu_options.allow_growth = True

sess = tf.Session(config=config)

# 设置 GPU 所使用的最大比例

config = tf.ConfigProto()

config.gpu_options.per_process_gpu_memory_fraction =0.7

config.gpu_options.allow_growth =True

sess = tf.Session(config=config)

查看信息版本安装路径

import tensorflow as tf

tf.__version__ # 版本

tf.__path_ # 路径

# __看着是一个下划线,实际上是两个下划线


转载:https://www.cnblogs.com/hellcat/p/6905878.html#_label0_9

相关文章

网友评论

      本文标题:TensorFlow的tips

      本文链接:https://www.haomeiwen.com/subject/qoursftx.html