安装
cuda10.1安装tensorflow-gpu的1版本会出现一些问题,应该在安装之前先输入
conda install cudatoolkit=10.0
在安装cudatoolkit之后再进行安装tensorflow就成功。
测试
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))
print(tf.test.is_gpu_available())
print(tf.__version__)