tensorflow cpu版本转gpu版本(win10)

时间:2024-04-04 13:00:19

动机

一开始因为图方便只装了cpu版的,后来在训练一个简单的一维CNN的时候发现每个step要200多ms,于是决定换成gpu版。

删除当前的tensorflow

之前没装过tensorflow的可以忽略这个步骤

先用pip list查看自己机器里现有的tensorflow组件
tensorflow cpu版本转gpu版本(win10)
用pip uninstall将所有跟tensorflow有关的删掉

安装CUDA和cuDAA

直接去官网下载
CUDA:CUDA下载
cuDAA:cuDAA下载
这里要注意CUDA版本和你要安装的tensorflow版本以及机器里的python版本相匹配,匹配关系可以看下图
cuDAA下好解压,会有bin、include、lib三个文件夹,将里面的东西复制到CUDA安装目录下对应的bin、include、lib文件夹里
详细过程可以参照这位博主的步骤
CUDA安装,作者jer8888
tensorflow cpu版本转gpu版本(win10)
如果不匹配在import tensorflow的时候会报下图的错,让你把dll文件加入环境变量。其实并不需要加,只要版本匹配装好之后可以直接使用。
tensorflow cpu版本转gpu版本(win10)

安装tensorflow-gpu

直接用pip安装,速度慢的话加选项 -i 去镜像站下载,我这里用的是清华的站,网上还可以找到其他的。记得在tensorflow-gup后面加上你要的版本号,不然会默认下载最新的。我的python版本是3.5,tensorflow-gpu版本是1.13.1(下图的1.4.0版本需要的CUDA版本是8.0,现在官网已经不能下载了),CUDA版本是10.0,亲测这个搭配可以使用!!安装完之后就可以在python里面使用了。
tensorflow cpu版本转gpu版本(win10)

效果

同样一个模型,每一个step所需的时间从200多ms降到了28ms,实现了质的飞跃。960的效果感觉还不错。