最近在学习深度学习,然后在安装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卷积神经网络的人脸识别程序,如果感兴趣的话可以看看。