Windows安装TensorFlow详细教程-2017-10-1

步骤:

1)安装anaconda 4.2.0(对应于python 3.5.2版本),因为TF支持的是python 3.5版本,所以安装这个版本可以省事,不然安装最新的python版本,需要再创建python 3.5的环境,蛋疼。这里放个清华镜像:https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/,下载速度杠杠的。

2)安装CUDA 8.0(在此之前,需要安装vs2015)

在官网下载cuda 8(有个补丁)以及cuDNN 6

Windows安装TensorFlow详细教程-2017-10-1_第1张图片

安装完之后系统变量会自动为你添加上,这个不用管。

测试一下是否安装成功,命令行输入nvcc -V,看到版本信息就表示安装成功了。

Windows安装TensorFlow详细教程-2017-10-1_第2张图片

3)安装 cuDNN

Google 于 2017 年 8 月 17 日左右发布了 TensorFlow 1.3,该版本不再支持 cuDNN 5,开始支持 cuDNN 6,并预计在 TensorFlow 1.4 支持 cuDNN 7,所以说当你使用 pip 安装最新版的时候,请使用 cuDNN 6,而不是我提供的 5.1,否则会出现Issues #2的问题。

TensorFlow 1.3 中关于 cuDNN 的发行说明:

All our prebuilt binaries have been built with cuDNN 6. We anticipate releasing TensorFlow 1.4 with cuDNN 7.

TensorFlow 1.2 中关于 cuDNN 的发行说明:

TensorFlow 1.2 may be the last time we build with cuDNN 5.1. Starting with TensorFlow 1.3, we will try to build all our prebuilt binaries with cuDNN 6.0. While we will try to keep our source code compatible with cuDNN 5.1, it will be best effort.

其实这个是一个压缩包,解压放到任何一个目录下就行,然后把你放的那个目录添加到Path环境变量里。我这里是将此解压缩包放在了cuda安装的文件夹下面:


Windows安装TensorFlow详细教程-2017-10-1_第3张图片
Windows安装TensorFlow详细教程-2017-10-1_第4张图片

上图解压缩后直接放在下图目录下覆盖即可,不用配置啥环境变量。

4)安装TensorFlow

       # GPU版本

                 pip3  install--upgrade  tensorflow-gpu

       # CPU版本

                  pip3   install--upgrade   tensorflow

如需升级pip就按照提示升级吧

电脑配置确实很重要,原来3500买的电脑,跑个简单的gan程序都要好几个小时(还有时卡死),下载16000的电脑只需几分钟,差距啊。

你可能感兴趣的:(Windows安装TensorFlow详细教程-2017-10-1)