win10安装tensorflow2.1 gpu版本

win10安装tensorflow2.1 gpu版本

在anaconda prompt里依次输入以下命令即可成功安装

conda create -n TF2.1 python=3.7 （创建虚拟环境）
y

conda activate TF2.1 （激活虚拟环境）

conda install tensorflow-gpu==2.1.0 （安装tensorflow2.1）

安装完成后

gpu测试程序（在pycharm里测试）
import tensorflow as tf

tensorflow_version = tf.version
gpu_available = tf.test.is_gpu_available()

print(“tensorflow version:”, tensorflow_version, “\tGPU available:”, gpu_available)

a = tf.constant([1.0, 2.0], name=“a”)
b = tf.constant([1.0, 2.0], name=“b”)
result = tf.add(a, b, name=“add”)
print(result)

结果
（前面一堆红色的）
tf.Tensor([2,4],shape=(2,)dtype=float32)