Win10下安装GPU版Tensorflow

查看显卡支持与否

查询GPU是否支持CUDA - 计算机视觉小菜鸟的专栏 - CSDN博客
http://blog.csdn.net/carson2005/article/details/46362277

0. 安装显卡驱动

如果计算机上已经有了(好像可以跳过),如果已经安装过了最新版本,看这里:
如何在 Windows 系统玩 TensorFlow
http://mp.weixin.qq.com/s?__biz=MzI2MzYwNzUyNg==&mid=2247483793&idx=1&sn=fc53b4f078d01417e6d152c5bf0b1faf&chksm=eab80418ddcf8d0eb7b9cb9f2f9234b0e2c2caf48fce49891e4ef0f338b913f7a13d0cc54629&mpshare=1&scene=23&srcid=0805B6qFiG0Kzj6TIdkCwsDs#rd

1. CUDA 8.0下载和安装

官网下载(需登录) :https://developer.nvidia.com/rdp/cudnn-download
注意一定要自定义安装!因为CUDA 8.0下载和安装的里面又重新安装了驱动,附下载地址cudnn-8.0-windows10-x64-v5.1.zip

2. 测试CUDA安装结果

打开命令提示符,输入:nvcc -V ,可以看到如下信息:
这里写图片描述
c:\ProgramData\NVIDIA Corporation\CUDA Samples\v8.0\bin\win64\Debug目录下找到deviceQuery.exe这个文件。打开一个cmd窗口,定位到 c:\ProgramData\NVIDIA Corporation\CUDA Samples\v8.0\bin\win64\Debug目录,输入:deviceQuery.exe ,然后回车。会得到如下结果:(为什么我这里是debug)
这里写图片描述

3.安装Tensorflow

  • 安装Anaconda
    Anaconda包含很多科学计算的包,例如numpy、scipy等等,可以为你省去不少麻烦,它的官网在这里:https://www.continuum.io/downloads ,下载最新的Anaconda 4.2版本,Python版本选择3.5 64bit的那个版本。
    嫌太慢,还可以去清华镜像:
    Index of /anaconda/archive/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror
    https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
    • 安装时,在Install for那个界面,建议选择【All Users(requires admin privileges)】那个选项。
    在如下那个界面中,把两个选项都勾上(默认是勾上的)
    这里写图片描述

任务管理器里输入:

Pip install tensorflow-gpu==1.0

更新操作:

python -m pip install –upgrade pip

建议不要更新,现在是TensorFlow 1.3,会有许多坑。因为Google 于 2017 年 8 月 17 日左右发布了 TensorFlow 1.3,该版本不再支持 cuDNN 5,开始支持 cuDNN 6,并预计在 TensorFlow 1.4 支持 cuDNN 7,所以说当你使用 pip 安装最新版的时候,请使用 cuDNN 6,而不是我提供的 5.1

4.安装cudnn

  1. cuDNN可以在前面GPU加速基础上大概再提升1.5倍的速度,它由nVIDIA开发。可以到nVIDIA官网上下载。(cuDNN Download | NVIDIA Developer
    https://developer.nvidia.com/rdp/cudnn-download)下载之前需要注册,然后问一系列问题,请耐心弄完。然后就可以下载了。不要下载错了,下载windows 10系统下64位的,最新的支持CUDA 8.0的cuDNN-5.1,文件名是:cudnn-8.0-windows10-x64-v5.1.zip
    2.下载完成后解压缩。里面有bin、include、lib三个目录,将三个文件夹复制到安装CUDA的地方覆盖对应文件夹,默认文件夹在:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0
    3.然后发现 GitHub 上果然有人提交了一个一样的 issue,大家可以在这里看下,然后是这么解决这个问题的:将下面这些文件复制到相应位置,
  2. C:\cuda\bin\cudnn64_5.dll —> C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin
  3. C:\cuda\include\cudnn.h —> C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\include
  4. C:\cuda\lib\x64\cudnn.lib —> C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib\x64
    除此之外,还有一个办法就是将C:\cuda\bin 也加进Path 环境变量里,经过测试这样也是可行的。
    这里写图片描述
    打开Anaconda Prompt,输入python,再输入import tensorflow,显示的如果是下图这样子,不提示没有安装cudnn,就成功了。
    这里写图片描述

参考

【1】【TensorFlow】Windows10 64位下安装TensorFlow - 官方原生支持 - Alan Lee - 博客频道 - CSDN.NET
http://blog.csdn.net/u010099080/article/details/53418159
【2】windows 10 64bit下安装Tensorflow+Keras+VS2015+CUDA… - 简书
http://www.jianshu.com/p/c245d46d43f0
【3】Windows10 TensorFlow “Couldn’t open CUDA library cudnn64_5.dll” - 博客频道 - CSDN.NET
http://blog.csdn.net/suo_ivy/article/details/70445103
【4】window10 + cuda8.0+tensorflow 安装教程 - DrSlump的博客 - CSDN博客
http://blog.csdn.net/drslump/article/details/54773818
【5】win10+cuda8.0+cudnn+Tensorflow(GPU)安装 - CSDN博客
http://blog.csdn.net/weixin_36368407/article/details/54177380
【6】Windows 10 +Anaconda+tensorflow+cuda8.0 环境配置 - CSDN博客
http://blog.csdn.net/u012223913/article/details/72420890

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

何以问天涯

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值