tensorflow cpu版本转gpu版本(win10)
2020-06-06 06:29
239 查看
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的效果感觉还不错。
相关文章推荐
- win10系统Anaconda和Pycharm的Tensorflow2.0之CPU和GPU版本安装教程
- Win10下安装GPU版本的tensorflow
- win10下基于python(anaconda)安装gpu版本的TensorFlow以及kears深度学习框架
- TensorFlow在win10下安装CPU版本
- 如何在Linux 系统 以及windows 系统安装 tensorflow -cpu -gpu 老版本 0.12 ,0.10
- 用Anaconda 卸载tensorflow(cpu版本)导入tensorflow-gpu (gpu版本) ----太方便了趴
- win10环境,基于Anaconda3(内置python3.7.4)安装tensorflow2.0.0(CPU版本),并在tensorflow环境下安装spyder
- tensorflow-gpu版本以及cpu版本下载链接
- win10 安装tensorflow GPU版本
- win10下安装anaconda3+python3.7+tensorflow2.0.0(GPU)+cuda10.0+cudnn+pytorch及查看cuda+cudnn版本
- Tensorflow 2.1 安装(CPU版本+GPU版本)
- 【Deep Learning】win10下配置tensorflow1.4(gpu版本)
- Win10 配置tensorflow1.3 GPU版本以及tensorboard的使用
- tensorflow的安装教程(Anaconda安装 win10 64位 cpu版本)
- win10安装tensorflow无GPU版本
- Win10安装cpu版本tensorflow报错ERROR: Exception: Traceback (most recent call last):
- win10下安装TensorFlow-GPU版本
- 64位WIN10通过anaconda安装CPU版本tensorflow
- Win10 安装Tensorflow-GPU版本
- win10内置Linux子系统安装GPU版本TensorFlow