window7中keras使用tensorflow-gpu版本

在windows7中,我装的环境为:

CUDA:8.0

  安装教程:64位Windows 7 安装配置CUDA7.5/8.0+cuDNN5.0图文详解

  说明:笔者在装完CUDA后,运行nvcc  -V没有问题,但Samples_vs2013编译时出现问题,一直没有解决,但最后也不影响跑深度学习时gpu的使用。

cuDNN:6.0

  这里特征注意,CUDA和cuDNN要版本匹配,在装cuDNN时,先要注册,然后在以前版本中看到cuDNN和CUDA对应的版本。

安装步骤:

  1. 下载之后解压缩,将.h .lib 和.dll 文件分别拷贝到cuda的include, lib/x64, bin 文件夹下
  2. 将cuDNN中bin,include和lib,添加到环境变量中

python版本是3.6.6

  装python后再更新pip,没有使用Anaconda,直接装各种需要的包

    pip install jupyter

    pip install tensorflow-gpu==1.3,  1.3版本

一些错误并解决:

  1.运行import tensorflow

  出现 .dll文件 错误,笔者是因为cuda和cuDNN版本不匹配,将版本改正后,就可以了。

  2.运行神经网络时,关键词:softmax  axis 错误

  将keras版本改为老版本 pip install keras==2.0.8,  以前版本为2.2.4

  3.如果运行神经网络时gpu使用率为0%,则说明没有使用,这时要注意以上步骤有没有出错

参考:

tensorflow-gpu和CUDA以及cuDNN版本匹配安装经验

tensorflow-gpu以及cuda、cudnn的安装——Linux

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值