win7 64位 安装tensorflow GPU版本

    近日终于在win7上安装成功tensorflow GPU版本,并测试成功!这里将其中遇到的种种问题和解决方法做下记录,以供大家相互交流!

    电脑配置:16G内存;处理器: AMD Athlon(tm)x4 870K ,3.9GHz;GTX1050显卡;win7 64bit系统

    平台搭建:python3.5.2(64bit)+cuda8.1+cuDNN5.1+tensorflow 0.12.0rc0.0(想升级以后可以随便升级)

    安装过程:

     第一步:安装python3.5.2

                 访问https://www.python.org/downloads/ 找到相应的版本,下载安装就可以了,可以选择自动添加path,也可以手动添加。python安装不多赘述,百度 到处都是。安装完成后打开cmd,输入python,查看是否安装成功,成功之后进入第二步。

    第二步:安装cuda8.1

                访问https://developer.nvidia.com/cuda-downloads,同样找到相应的版本下载即可,也很简单,下载之后执行.exe文件,默认步骤安装就可以。安装完 成后,打开cmd,输入:nvcc -V,即可查看。

    第三步:安装cuDNN:

            访问https://developer.nvidia.com/cudnn(要注册Nvidia用户,并加入CuDNN开发组,填若干问卷就可以下载了)选择下载版本时要注意和Cuda版本匹 配。下载完成后,解压,会看bin,lib,include三个文件,然后分别将这三个文件中的.dll文件复制到 C:\ProgramFiles\NVIDIA GPUComputingToolkit\CUDA\v8.0对应的文件夹内。                 

    第四步,安装tensorflow:

     在浏览器中打开https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-0.12.0rc0-cp35-cp35m-win_amd64.whl,下载tensorflow安装包到d盘(随便什么位置都行,只要后面对应好安装路径就可以),然后打开python安装目录下的Script文件将,将pip.exe拖入到cmd中,然后输入 install d:\tensorflow_gpu-0.12.0rc0-cp35-cp35m-win_amd64.whl,回车进行安装。


    到这里已经安装成功了,但是运行简单的程序可以,但是运行需要GPU加速的程序时就会出现问题,这个时候不要害怕,装个VS2015,问题就解决了。这个时候就可以测试了,可以去githun下个mnist的程序跑一下,会发现又有问题出现了,不要着急,可能是因为GPU内存的问题,需要更改batch,可以访问http://stackoverflow.com/questions/39076388/tensorflow-deep-mnist-resource-exhausted-oom-when-allocating-tensor-with-shape,查看修改代码就可以了。最后说明一下,我使用的是pycharm,各种包、模块装起来都比较方便。

  


   大家有问题随时欢迎交流。

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值