一、安装环境
TensorFlow即可以支持CPU,也可以支持CPU+GPU。前者的环境需求简单,后者需要额外的支持。TensorFlow是基于VC++2015开发的,所以需要下载安装VisualC++ Redistributable for Visual Studio 2015 来获取MSVCP140.DLL的支持。如果要安装GPU版本(有N卡,即NVIDIA显卡),需要以下额外环境:
0)有支持CUDA计算能力3.0或更高版本的NVIDIAGPU卡。
1)下载安装CUDA Toolkit 8.0,并确保其路径添加到PATH环境变量里;
2)下载安装cuDNN v6或v6.1,并确保其路径添加到PATH环境变量里;
3)CUDA8.0相关的NVIDIA驱动。
二、安装过程
具备了上述安装条件(CPU或GPU)之后,TensorFlow可以通过两种方式进行安装。一是”native” pip,二是Anaconda。推荐前者,比较简单。
一)使用”native” pip安装
1)下载安装Python 3.5.x 64-bit,要注意版本3.5.x。
2)打开windows的命令行窗口,安装CPU版本输入
pip3 install --upgrade tensorflow
安装GPU版本输入
pip3 install --upgrade tensorflow-gpu
二)使用Anaconda安装