系统环境:Windows 10 64bit
显卡型号:NVIDIA GEOFORCE 940M
显存大小:2G
破顽石FL5900U能跑TensorFlow-GPU我也是很开心的
-
安装Anaconda 3。下载地址:https://www.anaconda.com/download/#windows
默认安装即可。在Advanced Options处勾选Add Anaconda to my PATH environment variable -
安装CUDA v8.0。下载地址:https://developer.nvidia.com/cuda-80-ga2-download-archive
根据自己的环境选择对应的版本,exe分为网络版和本地版,网络版安装包比较小,执行安装的时候再去下载需要的包;本地版安装包是直接下载完整安装包。建议下载本地版。
楼主在这里分享个win10-v8.0版本:
https://pan.baidu.com/s/1Ihx6nhZ3ioASHAS8rBCwzQ
安装完成后,打开控制命令行,输入nvcc –V查看版本信息。
NVIDIA无法正确安装的解决方案:
楼主在安装过程中碰到这个问题= =好多次!!!几乎每个笔记本都碰到过。按网上说的升级vs、以管理员方式运行都不管用,后来还是在百度经验中找到答案。
参考链接:https://jingyan.baidu.com/article/f25ef25461affd482c1b82a6.html
我来做个搬运工:
安装前,右键“计算机”,点击“管理”。
点击“服务和应用程序”,再点击“服务”。
找到“Windows Installer”,右键选择“启动”并等待服务启动。 -
下载cuDNN。地址:https://developer.nvidia.com/cudnn。下载完成后解压,用解压得到的文件替换CUDA安装路径(我的是默认路径C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0)下的同名文件夹。由于下载cudnn需要注册,而且密码还超麻烦(= =大写+小写+数字,楼主只能记得纯数字的密码好吗,反正每次都忘记o(╯□╰)o),所以我还是给个我下载的版本链接,友情提示,本链接仅适用于64位win10,cuda8.0。
cudnn7: https://pan.baidu.com/s/19PiA3DpWD1GD5MPDrhh0NA
cudnn5: https://pan.baidu.com/s/1F2-wmU7-yz6qPMLC3S6Y4A
在此附上tensorflow-cuda-cudnn-python版本对应表,转载自博客https://blog.csdn.net/omodao1/article/details/83241074
替换完成后,打开 c:\ProgramData\NVIDIA Corporation\CUDA Samples\v8.0 目录下的CUDA的示例程序(这个文件夹是隐藏的,要先显示出来。里面对VS2013、VS2015各有生成解决方案),将解决方案配置更改为Release和x64,若编译成功,则CUDA和cuDNN安装成功。如果提示是缺少:“d3dx9.h”、“d3dx10.h”、"d3dx11.h"头文件,安装下DXSDK_Jun10.exe,下载地址:https://www.microsoft.com/en-us/download/confirmation.aspx?id=6812
再次编译。emmmmm经过多次测试,每次都会报这个错,所以建议直接先装DXSDK再编译吧。。
- 安装Tensorflow。保持网络连接,从开始菜单中打开Anaconda Prompt,输入:pip install tensorflow-gpu==1.2.0(如果不输入版本号,会默认安装最新版本1.3.x,但是我的CUDA是v8.0,会提示版本不符合要求)。等待一段时间,安装完成后测试Tensorflow是否安装成功,测试方法:在Spyder里输入下面的代码,不报错即可。
import tensorflow as tf
hello = tf.constant(‘Hello, TensorFlow!’)
sess = tf.Session()
如果报错的话。。。比如下面这样
在网上找了一些解决方案,有的是要装vs2017运行时,有的是要重装TensorFlow和anaconda,楼主都试过了,折腾了两天后来想起来在另外一台电脑上装的时候,装的cudnn5+tensorflow-gpu1.2.0,现在装cudnn7+tensorflow-gpu1.2.0就会报错,所以python版本控制很重要啊啊啊。。。换成cudnn5以后就好了。
- 安装Keras。保持网络连接,从开始菜单中打开Anaconda Prompt,输入:pip install keras,等待下载安装。