tensorflow
文章平均质量分 65
Mr丶Caleb
中国科学技术大学在读研究生
展开
-
tensorflow gpu使用说明
由于tensorflow默认抢占服务器所有GPU显存,只允许一个小内存的程序也会占用所有GPU资源。下面提出使用GPU运行tensorflow的几点建议:1.在运行之前先查看GPU的使用情况:$ nvidia-smi # 查看GPU此时的使用情况 或者$ nvidia-smi -l # 实时返回GPU使用情况2.目前实验室服务器有0,1,2,3四个GPU,找到空闲的GP原创 2016-10-22 15:32:44 · 53097 阅读 · 6 评论 -
tensorflow学习笔记四——实现一个CNN网络
本例子用到了minst数据库,通过训练CNN网络,实现手写数字的预测。首先先把数据集读取到程序中:mnist = input_data.read_data_sets('MNIST_data', one_hot=True)然后开始定义输入数据,利用占位符# define placeholder for inputs to networkxs = tf.placeholder(tf.float32,原创 2016-09-23 19:08:58 · 15236 阅读 · 10 评论 -
tensorflow建立一个简单的神经网络
本笔记目的是通过tensorflow实现一个两层的神经网络。目的是实现一个二次函数的拟合。如何添加一层网络代码如下:def add_layer(inputs, in_size, out_size, activation_function=None): # add one more layer and return the output of this layer Weights =原创 2016-09-23 16:04:33 · 16356 阅读 · 6 评论 -
tensorflow可视化tensorboard
本笔记主要讲如何运用tensorflow看自己创建网络的结构。可以通过with tf.name_scope(‘weights’):来添加模块的名称,并且with可以嵌套。对于变量也可以通过name参数来定义名字。然后在终端里输入 tensorboard –logdir=logs,再在浏览器里打开相应地址即可。from __future__ import print_functionimpor原创 2016-09-23 16:36:05 · 5094 阅读 · 0 评论