win10下tensorflow的安装指南(GPU)

此文章旨在作者本人保存一下已经调通了的安装步骤,避免日后再次掉坑

同时也希望借此文帮助正在坑里挣扎或者即将掉坑的小白同胞们

(创作日期为2017年5月,如果是很多个月后才看到此文的小伙伴就不建议继续看下去了,估计会有新坑出现。。。)

题主电脑的GPU是NVIDIA 的TITAN X系列,支持CUDA加速,所以本文主要针对GPU版本的安装(我也安装过CPU-ONLY的版本,个人感觉比GPU版本坑少,也欢迎一起讨论)

至于大家的电脑是否支持CUDA需要在NVIDIA的官网查询一下(直接百度应该也可以搜的到)

那么我们的绕坑之旅现在开始

1.安装Anaconda3

这里需要注意的是目前的仅支持python3.5所以需要必须下载对应的版本,链接在此

Anaconda installer archive

下载叫做 Anaconda3-4.2.0-Windows-x86_64.exe 的文件,千万不要下载错了哦

下载完成后双击exe文件即可安装

2.安装cuda8.0

在这里下载相应的CUDA版本,注意一定要是8.0版本哦,我选的是local版本

CUDA Toolkit Download

下载后双击exe文件安装

然后需要下载cudnn

NVIDIA cuDNN

这里要注意的是,安装版本一定要是v5.1,作者在这个坑里困了很久才跳出来

cudnn的安装方法很有趣,解压后会有很多文件夹,在之前安装好的cuda的根目录下,找到同样名字的文件夹,把cudnn里的文件拷到cuda的文件夹里

(感谢评论里的
@叶柒同学提供了另一种方法

“cudnn 6.0版本也可以,需要自己去CUDA下面把对应的dll文件改名为cudnn60_5.dll,然后就可以了。”)

安装全部完成后可以在cmd中输入

nvcc -V

来测试一下是不是安装成功,如果成功的话会显示版本号

3.升级pip

先升级pip的版本,升级的目的是为了防止之后的反复警告

python -m pip install -U pip

为了日后的pip获取资源不卡不迷之掉线(大天朝的墙你懂的)最好是在此顺便把软件获取源一同改掉

我们需要在user目录中创建一个pip目录,如:

C:\Users\你的用户名\pip

然后新建一个文件叫(后缀很重要,可以先建txt,更改内容后再改后缀为ini)

pip.ini

里面的内容如下 ,记得保存

[global]
index-url = https://pypi.tuna.tsinghua.edu.cn/simple

这样的话之后就是从清华获取源了

4.安装tensorflow

使用管理员模式运行cmd(可以在开始菜单里右键管理员模式打开)

输入代码

pip install --upgrade --ignore-installed tensorflow-gpu

默默的等待安装,注意看有没有报错

5.安装VS的支持模块

这是一个我陷进去很久的坑,如果没有装过VS的话会各种报错,解决方法是安装

(原链接已挂!待作者电脑修复好后附上网盘链接,不嫌麻烦的同学可以先安装VS全家桶)

不过为了保险起见,作者一并安装了VS2015全家桶

Free Dev Tools – Visual Studio Community 2013

6.跑个程序试试吧!

如果你顺利的进行到这一步,那么理论上来说你的电脑已经成功的安装了tensorflow

使用管理员模式打开cmd

然后输入python

接下来输入

import tensorflow as tf

如果没有报错的话安装就成功啦,你可以移步tensorflow的中文社区去试试helloworld的小代码并开启你的机器遗忘生涯
TensorFlow中文社区-首页

如果你发现了新坑也欢迎和作者讨论哦

    原文作者:Square Tsou
    原文地址: https://zhuanlan.zhihu.com/p/27092679
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞