在Windows下安装TensorFlow

时间:2022-05-04
本文章向大家介绍在Windows下安装TensorFlow,主要内容包括1.下载Python、升级pip、安装tensorflow、GPU版本、安装TensorFlow的GPU支持包、基本概念、基础应用、原理机制和需要注意的事项等,并结合实例形式分析了其使用技巧,希望通过本文能帮助到大家理解应用这部分内容。

目前TensorFlow在Windows下只支持Python 3.x。

1.下载Python

首先安装Pythone 3.6.4

下载地址

升级pip

python.exe -m pip install --upgrade pip

安装tensorflow

CPU版本

pip install --upgrade tensorflow

GPU版本

pip install --upgrade tensorflow-gpu

安装TensorFlow的GPU支持包

安装CUDA Toolkit 9.1

https://developer.nvidia.com/cuda-downloads

安装cuDNN 7.1

https://developer.nvidia.com/cudnn

$ python
...
>>> import tensorflow as tf
>>> hello = tf.constant('Hello, TensorFlow!')
>>> sess = tf.Session()
>>> print(sess.run(hello))
Hello, TensorFlow!
>>> a = tf.constant(1)
>>> b = tf.constant(2)
>>> print(sess.run(a + b))
3
>>>