TensorFlow小记

1. 安装TensorFlow

   conda install tensorflow-gpu
   

2. 测试TensorFlow是否安装成功

   import tensorflow as tf 
   sess = tf.Session() 
   a = tf.constant(1) 
   b = tf.constant(2) 
   print(sess.run(a+b)) 
   

   tensorflow 1.0版本的代码要创建会话Session
   需要注意的是：tensorflow的代码默认会占用所有显卡的所有显存

3. TensorFlow按需分配显存
   参考链接：tensorflow 使用 cpu 而不使用 gpu 问题 - 万道一 - 博客园 (cnblogs.com)

TensorFlow小记

原文：https://www.cnblogs.com/Todd-Qi/p/15042059.html