tensorflow gpu版安装(win10)
动机
一开始因为图方便只装了cpu版的,后来在训练一个简单的一维CNN的时候发现每个step要200多ms,于是决定换成gpu版。
删除当前的tensorflow
之前没装过tensorflow的可以忽略这个步骤
先用pip list查看自己机器里现有的tensorflow组件
用pip uninstall将所有跟tensorflow有关的删掉
安装CUDA和cuDAA
直接去官网下载
CUDA:CUDA下载
cuDAA:cuDAA下载
这里要注意CUDA版本和你要安装的tensorflow版本以及机器里的python版本相匹配,匹配关系可以看下图
cuDAA下好解压,会有bin、include、lib三个文件夹,将里面的东西复制到CUDA安装目录下对应的bin、include、lib文件夹里
详细过程可以参照这位博主的步骤
CUDA安装,作者jer8888
如果不匹配在import tensorflow的时候会报下图的错,让你把dll文件加入环境变量。其实并不需要加,只要版本匹配装好之后可以直接使用。
安装tensorflow-gpu
直接用pip安装,速度慢的话加选项 -i 去镜像站下载,我这里用的是清华的站,网上还可以找到其他的。记得在tensorflow-gup后面加上你要的版本号,不然会默认下载最新的。我的python版本是3.5,tensorflow-gpu版本是1.13.1(下图的1.4.0版本需要的CUDA版本是8.0,现在官网已经不能下载了),CUDA版本是10.0,亲测这个搭配可以使用!!安装完之后就可以在python里面使用了。
效果
同样一个模型,每一个step所需的时间从200多ms降到了28ms,实现了质的飞跃。960的效果感觉还不错。