安装TensorFlow

conda install tensorflow-gpu

测试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的代码默认会占用所有显卡的所有显存

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