win10下GPU加速的Keras配置(CUDA, cuDNN)

实践时重点参考了这一篇博客,但仍然遇到了不少问题,其中最普遍的是各个部分更新后的相互支持问题,本文只是实践成功的经历记录,在短时间内、环境相同的情况下应当比较有效,本文不会普遍性地解决安装CUDA、cuDNN中的各种问题
1. 安装conda、keras
conda install keras
2. 安装CUDA、添加环境变量。
a) 下载CUDA9.0版,当前(2018-3-21)tensorflow不支持最新的9.1
b) 安装时选择“精简”就行,不需要自定义安装
c) 添加下列环境变量
这里写图片描述
PATH环境变量
PATH变量
3. 下载cuDNN,选择7.0.5、for CUDA9.0版本
我之前安装了v7.1.1 for CUDA 9.1,tensorflow报错应当安装7.0开头的版本,后来发现原因应当是用了之前下载的for CUDA 9.1,但为了省事没有尝试使用v7.1.1 for CUDA 9.0
这里写图片描述
下载完后,将下图文件一起复制粘贴到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0,进行覆盖(应该不会有重复文件)
这里写图片描述
4. 测试安装是否完成
cd到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\extras\demo_suite目录下,执行bandwidthTest.exe,deviceQuery.exe,两步都打印出【Result=PASS】,则CUDA安装成功
5. 使用GeForce更新驱动,安装CUDA时应当已经安装了GeForce,如果没有可以去官网下载
我出现过驱动没更新导致tensorflow跑深度学习不成功的情况
6. 安装tensorflow-gpu
pip install --upgrade tensorflow-gpu
7. 命令行进入python环境
import tensorflow as tf
sess = tf.Session()

看打印信息最后几条,是否包括添加gpu,添加是否成功
8. 跑一个DeepLearning,看一下是否已加速
到上一布并不保证实际使用时不会出现问题

  • 2
    点赞
  • 11
    收藏
    觉得还不错? 一键收藏
  • 5
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值