安装TensorFlow 1.4版本 的环境要求 如图所示:
安装3.5以上版本的python,推荐直接安装Anaconda:https://www.continuum.io/downloads
CPU版本:
- 打开运行—-输入命令“cmd”—-输入命令“pip install tensorflow-cpu ”
GPU版本:
检查GPU是否支持CUDN:其实这一步是可以省略的,因为CUDN支持的显卡范围特别广,你基本只要确认你的GPU是不是NVIDIA的就可以了
下载CUDA:https://developer.nvidia.com/cuda-toolkit-archive,选择CUDA Toolkit 8.0,再选Windows,下载完之后打开安装,一路确认到底
下载cuDNN:https://developer.nvidia.com/rdp/cudnn-download,需要先注册了才能下载,选择Download cuDNN v6.0 (April 27, 2017), for CUDA 8.0,下载后将其解压缩就可以
设置环境变量:右键“我的电脑”,选择“高级系统设置”,进入“环境变量”,进入“Path”
添加”c:\cuda\bin”与”C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin”后点击确认打开运行—-输入命令“cmd”—-输入命令“pip install tensorflow-gpu ”,默认安装最新版,如果需要指定版本可以输入命令:“pip install tensorflow-gpu ==版本号”