随笔分类 - A-tensorflow
摘要:keras使用horovod多gpu训练 Horovod以类似的方式支持Keras和常规TensorFlow。要使用Horovod,请在程序中添加以下内容。 运行hvd.init()。 使用固定服务器GPU,以供此过程使用config.gpu_options.visible_device_list。
阅读全文
摘要:tensorflow使用horovod多gpu训练 要使用Horovod,在程序中添加以下内容。此示例使用TensorFlow。 运行hvd.init() 使用固定服务器GPU,以供此过程使用config.gpu_options.visible_device_list 通过每个进程一个GPU的典型设
阅读全文
摘要:import tensorflow as tf import os os.environ["CUDA_VISIBLE_DEVICES"] = '0' #use GPU with ID=0 config = tf.ConfigProto() config.gpu_options.per_process
阅读全文
摘要:打印pb模型参数及可视化结构import tensorflow as tf from tensorflow.python.framework import graph_util tf.reset_default_graph() # 重置计算图 output_graph_path = '/home/h
阅读全文
摘要:from tensorflow.python import pywrap_tensorflow import os checkpoint_path=os.path.join('./model.ckpt-100') reader=pywrap_tensorflow.NewCheckpointReade
阅读全文
摘要:输入一下代码即可 import tensorflow as tf print(tf.test.is_gpu_available())
阅读全文
摘要:tf.gather:用一个一维的索引数组,将张量中对应索引的向量提取出来 1 import tensorflow as tf 2 3 a = tf.Variable([[1,2,3,4,5], [6,7,8,9,10], [11,12,13,14,15]]) 4 index_a = tf.Varia
阅读全文
摘要:在mask中定义true,保留与其进行运算的tensor里的部分内容,相当于投影的功能。 mask与tensor的维度可以不相同的,但是对应的长度一定要相同,也就是要有一一对应的部分; 结果的维度 = tensor维度 - mask维度 + 1 以下是参考连接的例子,便于理解:
阅读全文
摘要:Tensorflow中的图(tf.Graph)和会话(tf.Session) Tensorflow编程系统 Tensorflow工具或者说深度学习本身就是一个连贯紧密的系统。一般的系统是一个自治独立的、能实现复杂功能的整体。系统的主要任务是对输入进行处理,以得到想要的输出结果。我们之前见过的很多系统
阅读全文
摘要:libcublas.so.9.0: cannot open shared object file: No such file or directory 输入命令: sudo ldconfig/usr/local/cuda-9.0/lib64 tensorflow.python.framework.e
阅读全文
摘要:tensorflow框架整体结构 用张量tensor表示数据;计算图graph表示任务;在会话session中执行context; 通过变量维护状态;通过feed和fetch可以任意的操作(arbitrary operation)、赋值、获取数据 TensorFlow的基本使用(基本结构、变量管理、
阅读全文