Windows下CUDA + TensorFlow-gpu + Keras 配置的坑

最近在学习深度学习,然后在安装CUDA,TensorFlow-gpu,Keras时遇到了一些坑...下面是我遇到的一些问题

首先我python这方面,是安装的Anaconda3自带的python v3.6.3,其实这里也有个坑,顺便说一下。在10月份的时候试了一下,windows下如果你安装Anaconda3最新版,那么自带python版本是v3.7,你可能装不了TensorFlow,当然现在不知道能不能,因为这个一直是在改进兼容的。(机器学习和深度学习,还是不推荐用自己下载的python版本,要安装各种各样的库简直不要太麻烦,亲身体会。。。所以还是安装Anaconda比较好。)

Anaconda这方面就不多说,进入正题,首先先说一下我的安装配置:

显卡:Nvidia GeForce GTX970M

CUDA :8.0

TensorFlow-gpu:1.4.0

cuDNN:6.0

Keras :2.1

首先说一下,这些配置的版本,对应非常重要,如果你想安装最新版,可以去百度或者google查看相对应的版本。而且如果你的显卡不是很好的话,最好查看一下你的显卡是否支持你想安装的CUDA版本。

说一下我遇到的问题吧,首先在这里先贴一下别人的安装教程,我安装CUDA就是在这里看的Windows10配置TensorFlow-GPU及Keras教程   顺便说下,你如果用nvcc -v 查看不了CUDA版本,那就试一下 nvcc --version

一、CUDA和TensorFlow版本冲突:

我按照这个链接里的教程,安装了CUDA v8.0,一开始我装的是cuDNN v7.0(这个先别装,后面会说明版本冲突)

在安装完CUDA和cudnn后,还没有出问题,就是在安装TensorFlow-gpu的时候,就出问题了,因为你如果用

pip install tensorflow-gpu下载的话,默认给你下载最新版,当前是1.12.0,但是CUDA v8.0和tensorflow-gpu v1.12.0版本不支持,然后只能将tensorflow-gpu退回1.4的版本,因为安装CUDA比较麻烦,所以我就降低了tensorflow-gpu版本,

pip install tensorflow-gpu==1.4

如果你想用版本高一点的tensorflow-gpu的话,那你要安装高版本的CUDA了,但是要看一下你的显卡支不支持了。

二、cuDNN和TensorFlow版本冲突:

如果你进入cuDNN安装那里,你会发现cuDNN v7.0,是适用于CUDA v8.0的对吧,我当初就是看到这个然后直接点击安装了,结果在运行程序的时候,报错说找不到cuDNN v6.0的一些文件....原谅我没有随时随地记录我bug报错信息的习惯。后来查了一下,发现是和TensorFlow-gpu的版本有关。所以cuDNN最好安装 v6.0,如果你是安装CUDA v8.0的话。

三、keras版本问题:

这个其实比较特殊,可以注意到我上面说的是版本问题,而不是版本冲突。我当初试着运行了一个基于CNN的人脸识别的程序,就有用到keras(后面会另发文章,这也是我为什么安装这些配置)。

然后那个程序在跑了以后出现了一个bug,网上查了一下,说是keras版本问题(开始装的是v2.2.4),然后要我换成v2.1,于是我只好再次退版本 pip install keras==2.1  关掉pycharm重新打开运行,没有bug。但是 奇怪的事情发生了,我重新看了一下项目包,我的keras版本还是2.2.4(如果有哪个大哥知道原因请告诉我下)。

 

如果你在pip install的时候,出现一大串红色的错误信息,那么试一下更新pip版本,红色报错信息下面有个黄色提示信息应该有的。 如果还是不行,然后你又安装了anaconda的话,就试试 conda install pip,再不行就百度一下看看别人有没有遇到相同的问题吧。

以上就是我遇到的一些问题,如果有错误的地方,请指正,感谢! 顺便说下,接下来几天我会发基于CNN卷积神经网络的人脸识别程序,如果感兴趣的话可以看看。

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值