TensorFlow
文章平均质量分 60
sherry颖
这个作者很懒,什么都没留下…
展开
-
TensorFlow学习笔记6:神经网络搭建(layer,estimator等)
这次来看一看如何用layer等API来搭建神经网络。from __future__ import print_function# Import MNIST datafrom tensorflow.examples.tutorials.mnist import input_datamnist = input_data.read_data_sets("/tmp/data/", one_ho原创 2017-11-28 14:57:00 · 743 阅读 · 0 评论 -
tensorflow指定GPU与动态分配GPU memory
在tensorflow中,默认指定占用所有的GPU,如需指定占用的GPU,可以在命令行中:export CUDA_VISIBLE_DEVICES=1这样便是只占用1号GPU,通过命令nvidia-smi可以查看各个GPU的使用情况。另外,也可以在python程序中指定GPU,并且动态分配memory,代码如下import osimport sysos.environ['CUDA_VISIBLE...原创 2018-03-15 11:02:50 · 4027 阅读 · 0 评论 -
TensorFlow学习笔记5:神经网络搭建
代码来源:https://github.com/aymericdamien/TensorFlow-Examples首先看一下用最基本的矩阵运算来搭建神经网络解决MNIST分类问题,比如搭建一个两层的全连接网络。from __future__ import print_function# Import MNIST datafrom tensorflow.examples.tutoria原创 2017-11-28 13:56:08 · 1871 阅读 · 0 评论 -
TensorFlow学习笔记3:线性回归
代码来源:https://github.com/aymericdamien/TensorFlow-Examples/'''A linear regression learning algorithm example using TensorFlow library.Author: Aymeric DamienProject: https://github.com/aymericdami原创 2017-11-26 11:03:43 · 365 阅读 · 1 评论 -
TensorFlow学习笔记2:基本运算
import tensorflow as tfa = tf.constant(2)b = tf.constant(3)with tf.Session() as sess: print "a: %i" % sess.run(a), "b: %i" % sess.run(b) print "Addition with constants: %i" % sess.run(a+b原创 2017-11-26 10:33:39 · 1629 阅读 · 0 评论 -
TensorFlow学习笔记1:Hello TensorFlow!
根据 https://github.com/aymericdamien/TensorFlow-Examples 学习TensorFlowimport tensorflow as tfhello = tf.constant('Hello, TensorFlow!')sess = tf.Session()print(sess.run(hello))其中第二行定义了常量hello,在Ten原创 2017-11-26 10:14:11 · 433 阅读 · 0 评论 -
TensorFlow学习笔记9:LSTM搭建
用TensorFlow来搭建LSTM,这里搭建LSTM来进行MNIST回归。对于MNIST分类问题,28*28的图片被LSTM视为像素序列,从而进行分类。代码来源:https://github.com/aymericdamien/TensorFlow-Examples首先,设置基本的参数from __future__ import print_functionimport tens原创 2017-11-29 10:35:02 · 3855 阅读 · 0 评论 -
TensorFlow学习笔记8:CNN搭建(layer,estimator等)
同样的,学习一下用layer等API来搭建CNN。代码来源:https://github.com/aymericdamien/TensorFlow-Examples首先,设置相关参数。from __future__ import division, print_function, absolute_import# Import MNIST datafrom tensorflow.原创 2017-11-28 19:43:35 · 2047 阅读 · 0 评论 -
TensorFlow学习笔记7:CNN搭建
代码来源:https://github.com/aymericdamien/TensorFlow-Examples用TensorFlow来搭建CNN解决MNIST分类。from __future__ import division, print_function, absolute_importimport tensorflow as tf# Import MNIST dataf原创 2017-11-28 16:29:06 · 323 阅读 · 0 评论 -
ImportError: libcudnn.so.6: cannot open shared object file 解决方法
在运行tensorflow-gpu时遇到了这个问题,可以通过以下命令解决。export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64在执行命令之前可以先cd到目录下,看文件是不是在里面。...原创 2018-03-17 13:58:43 · 1895 阅读 · 1 评论