参考https://www.cnblogs.com/pgzhang/p/9146617.html
1 查看自己的显卡是不是独立显卡
2 本机显卡型号 nvidia gtx 960 m,系统为win8
3 安装python3.6.6(64位)
4 安装tensorflow--gpu pip 安装的是1.5版本的,我安装的是1.8 版本的,pip install tensorflow--gpu==1.8.0
注意版本问题:不要随便安装版本,不然会报错说找不到一些dll文件(ImportError: Could not find 'cudart64_90.dll'.),
tensorflow-gpu 1.7以及之后的版本要安装 CUDA 8.0 以上的版本,tf 1.7 之下的安装 CUDA 8.0 之下的 ,安装 cuDNN 时要根据 CUDA 的版本来选择
5 安装cuda 在官网下载 cuda9.0
大概是1.4g的样子,下载之后安装,由于装了360,会有很多的阻止提示,要一直点击允许,不要错过,有些阻止后不知道会不会报错
安装的时候可以自定义安装的文件,不要太长的路径,因为后续会把cudnn的文件复制过来,可能会报错说没有文件过程,不能复制的问题
安装玩之后,自定义的cuda的文件后,还会继续安装在c盘的program file文件加里的 有个nvidia gpu computing toolkit 文件
如果安装正确的话,会有(cuda》v9.0》bin、clude、lib等14的文件)
6 安装之后运行tensorflow 报错说找不到cudnn的dll文件,去官网继续下载cudnn ,版本嗯哼cuda保持一直,对应的9.0
7 讲cudnn的文件解压,有clude,bin,lib/64X,然后把里面的文件分别复制到cuda/v9.0里的相对应的文件里
8 再运行tensorflow 不会再报错
9 如果跑程序的时候报错:if you want to see a list of allocated tensors when oom happend....
减小数值或者bitchsize的大小