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)

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