-
安装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)