Windows 10 下安装Tensorflow1.0.1-GPU

Windows 10 下安装Tensorflow1.0.1-GPU

之前在自己的windows上安装了tensorflow1.0.1-CPU版,后来想用gpu进行计算,于是安装gpu版。没想到软件之间的依赖关系、版本等导致数种问题,百度谷歌良久才调通程序,特记下曲折的安装过程和一些细节,尽量解释选择软件版本的原因,希望能减轻读者的痛苦。

我的环境

  • 操作系统:Windows 10 64bit
  • 显卡型号:GeForce GTX 950M

软件准备

  • Anaconda3-4.2.0-Windows-x86_64.exe

    Anaconda是一个用于科学计算的Python发行版,安装后,python以及常用的用于计算的package如numpy、matplotlib、Pillow也都安装好了,而且还能切换不同版本的python,很方便。
    安装tensorflow是不必安装Anaconda的,但是你至少要有python环境,而且需要是python3.5.x,因为正式1.0版的tensorflow不支持python2.x版。
    Anaconda3中内置的python为3.x版本(Anaconda2中内置的python为2.x版本),不过最新的Anaconda内置python为3.6版,为了避免潜在的麻烦,我安装了较新的历史版本(没错,Anaconda可以切换python版本,不过我之前在Anaconda2下安装python3后出现了一些问题,没时间较真了)。

    Anaconda 官网历史版本下载
    一篇不错的入门教程

  • tensorflow_gpu-1.0.1-cp35-cp35m-win_amd64.whl

    tensorflow在github的提供了以下几个版本:
    Linux CPU-only: Python 2 / Python 3.4 / Python 3.5
    Linux GPU: Python 2 / Python 3.4 / Python 3.5
    Mac CPU-only: Python 2 / Python 3
    Mac GPU: Python 2 / Python 3
    Windows CPU-only: Python 3.5 64-bit
    Windows GPU: Python 3.5 64-bit
    Android: demo APK, native libs
  • DirectX SDK

    官网下载

  • VS2015/2013/2012

    安装cuda前必须先安装一款VS,具体安装哪款,请参照下面对于cuda的介绍。

  • cuda_8.0.61_win10.exe

    cuda是一种并行计算平台,可以利用gpu进行并行计算。
    请注意,tensorflow1.x-gpu版只支持cuda8.x版本你可以在这里确认这个问题

    操作系统对cuda8的支持(常用的系统应该都没问题,即使在cross情况下——32位的系统在64位机器上):

    Operating SystemNative x86_64Cross (x86_32 on x86_64)
    Windows 10YESYES
    Windows 8.1YESYES
    Windows 7YESYES
    Windows Server 2012 R2YESNO
    Windows Server 2008 R2 DEPRECATEDYESYES

    VS对cuda8的支持(注意cross情况):

    Compiler IDEOperating SystemNative x86_64Cross (x86_32 on x86_64)
    Visual C++ 14.0Visual Studio 2015YESYES
    Visual C++ 14.0Visual Studio Community 2015YESNO
    Visual C++ 12.0Visual Studio 2013YESYES
    Visual C++ 11.0Visual Studio 2012YESYES
    Visual C++ 10.0 DEPRECATEDVisual Studio 2010YESYES

    我之前就不幸地安装了唯一一种不支持的情况,安装Visual Studio Community 2015 32bit在我64bit笔记本上,报了一堆错,卸载后安装vs2013后解决问题。
    到这里你就可以根据自己的情况安装对应的VS了。

    cuda官网安装教程(这里有上面两张表格)
    官网下载

  • cudnn-8.0-windows10-x64-v5.1

    针对深度神经网络采用GPU计算的加速库,下载解压后你会得到3个.dll文件。

    官网下载(下载需要注册,并且回答一些问题)


一些问题

  • cudnn是必要的
    我之前参考了一些教程,以为cudnn只是为了提速,并不是必须的,就没有安装,结果出错。在tensorflow官网安装教程中,要求cudnn是必装的,并且要将动态链接库文件添加到环境变量中,我照做后仍报错,后来参照这篇教程将3个.dll文件拷贝到duda安装目录下对应路径下,解决问题。
  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值