近日终于在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,各种包、模块装起来都比较方便。
大家有问题随时欢迎交流。