tensorflow-gpu安装的常见问题及解决方案


Posted in Python onJanuary 20, 2020

装tensorflow-gpu的时候经常遇到问题,自己装过几次,经常遇到相同或者类似的问题,所以打算记录一下,也希望对其他人有所帮助

基本信息

  • tensorflow-gpu
  • pip安装(virtualenv等虚拟安装实质也是pip安装,只是建了个独立的环境,不会影响系统环境,查问题比较容易,最多重新再创建一个干净的环境再来)

安装完之后会用import tensorflow看是否安装成功,结果报错,主要有碰到下面两大类报错信息:

1.ImportError: DLL load failed: 找不到指定的模块 之pywrap_tensorflow.py

报错信息里面有大量的pywrap_xxx相关的脚本报错:

Traceback (most recent call last):
 File "E:\study\machinelearning\ENV\lib\site-packages\tensorflow\python\pywrap_tensorflow.py", line 58, in <module>
 from tensorflow.python.pywrap_tensorflow_internal import *
 File "E:\study\machinelearning\ENV\lib\site-packages\tensorflow\python\pywrap_tensorflow_internal.py", line 28, in <module>
 _pywrap_tensorflow_internal = swig_import_helper()
 File "E:\study\machinelearning\ENV\lib\site-packages\tensorflow\python\pywrap_tensorflow_internal.py", line 24, in swig_import_helper
 _mod = imp.load_module('_pywrap_tensorflow_internal', fp, pathname, description)
 File "E:\study\machinelearning\ENV\lib\imp.py", line 242, in load_module
 return load_dynamic(name, filename, file)
 File "E:\study\machinelearning\ENV\lib\imp.py", line 342, in load_dynamic
 return _load(spec)
ImportError: DLL load failed: 找不到指定的模块。

这类错误出现的最多,主要有几大类原因:

(1)Microsoft Visual C++ 2015 Redistributable Update 3 没有装

这个是自己第一次装的时候碰到的,下载 vc_redist.x64.exe 安装之后就ok了

再生波澜

自己今天再装的时候,下载下来发现安装不了,看日志是说我的vs版本比较新,所以不能装。这个时候可以可以看看自己本机的system32下面有没有MSVCP140.DLL这个文件

其他解决方案

有些网友说用的比较新的tensorflow,装了2017的Redistributable包就好了,你也可以试试

我再装完2017的包之后,并且检查自己系统中已经有了MSVCP140.DLL文件依旧报同样的错误

(2)cuda和cudnn版本不一致

这个问题也是非常多的,我装了很多次的cuda基本上没有安装失败过,但是遇到和cudnn版本不一致的情况。因为下载的cuda默认是最新版本的cuda10.0,而我下载的cudnn当时用的旧的,也就是给cuda9.0的,所以后面换了一下也就解决问题了

cuda下载

tensorflow-gpu安装的常见问题及解决方案

我这里默认点完自己系统的配置(win10x64)得到的是最新的cuda10-win10,可以点击最右边的Legacy Releases看到更早一点的版本

cuda安装和验证

一路next貌似没遇到过啥问题

验证的话:在命令行下面输入nvcc -V,看是否OK

另外sample下面的两个是deviceQuery.exe和bandwidthTest.exe执行都没有出现问题过

cudnn下载

要登录nvidia developer账号

tensorflow-gpu安装的常见问题及解决方案

点开最下面的Archived cuDNN Releases可以看到更多的版本,因为我下载的是cuda-9.0,稳妥起见,我下载的cudnn版本是:Download cuDNN v7.0.5 (Dec 5, 2017), for CUDA 9.0

按照道理来讲这里的Download cuDNN v7.5.0 (Feb 21, 2019), for CUDA 9.0应该也可以,下次验证再确认一下。

cudnn安装

在下载的页面可以打开Installation-Guide看一下windows的cudnn安装指南,主要有以下操作

(1)把解压缩的cudnn下面的bin、lib和include三个文件夹下面的文件拷贝到cuda安装的目录下面同名的目录下面

cuda路径:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0

(2)把CUDA路径添加到环境变量的CUDA_PATH中

tensorflow-gpu安装的常见问题及解决方案

cuda本书在安装的时候会把cuda的安装路径添加的环境变量的path中(注意:是在path的最前面,不容易看到),所以不必自己把cuda的路径添加到path中

这里自己是把解压后的cudnn放到d盘,比如:D\cuda,然后把D:\cuda\bin放到了path中,因为网上有些人是这样建议的。但是看cudn的安装指南并没有提及到,所以感觉应该不需要

很遗憾的是,今天保证这里版本一直之后,还是依旧报 = =

(3)tensorflow-gpu版本不一致

安装tensorflow-gpu的时候一般都是用的默认指令:

pip install --upgrade tensorflow-gpu

结果是会把tensorflow-gpu的最新版本装上,我的版本情况如下:

(1)python:3.6.0 (2)cuda-9.0 (3)cudnn-7.0 (4)tensorflow-gpu-1.13.0

最新的cuda是10.0了,但是我装的是9.0,所以我把tensorflow-gpu装到1.12.0,然后完美解决问题了。_

pip uninstall 
tensorflow-gpu==1.13.0
pip install tensorflow-gpu==1.12.0

这里说明tensorflow-gpu1.13.0估计是用了最新的cuda版本中的内容,也算是版本不一致了。

如果跟我一样,上面的问题都解决了,那就看看是不是这里版本太新或者太旧了。这里有个插曲,因为我开始不小心把1.12.0输成了1.2.0,结果还是不行,没注意结果纯粹浪费了一段时间。

(4)其他python库版本问题等

网上有些人还遇到numpy等python库版本等的问题,我倒是没遇到,因为安装tensorflw-gpu的时候会把相关的依赖包都给下载下来

2.TensorFlow pip installation issue: cannot import name 'descriptor'之graph_pb2.py

报错信息如下有graph_xxx相关的脚本报错:

Traceback (most recent call last):
 File "<stdin>", line 1, in <module>
 File "F:\study\machinelearning\ENV\lib\site-packages\tensorflow\__init__.py", line 24, in <module>
 from tensorflow.python import pywrap_tensorflow # pylint: disable=unused-import
 File "F:\study\machinelearning\ENV\lib\site-packages\tensorflow\python\__init__.py", line 59, in <module>
 from tensorflow.core.framework.graph_pb2 import *
 File "F:\study\machinelearning\ENV\lib\site-packages\tensorflow\core\framework\graph_pb2.py", line 6, in <module>
 from google.protobuf import descriptor as _descriptor
 File "F:\study\machinelearning\ENV\lib\site-packages\google\protobuf\descriptor.py", line 47, in <module>
 from google.protobuf.pyext import _message
ImportError: DLL load failed: 找不到指定的程序。

这个我碰到过两次,都是protobuf的版本高了的缘故,网上搜到的也是这个原因,把protobuf的版本从3.6.1降到3.6.0解决

pip list
pip uninstall protobuf
pip install protobuf==3.6.0
pip list

参考

[1]import error: load dll failed

总结

以上所述是小编给大家介绍的tensorflow-gpu安装的常见问题及解决方案,希望对大家有所帮助,如果大家有任何疑问请给我留言,小编会及时回复大家的。在此也非常感谢大家对三水点靠木网站的支持!如果你觉得本文对你有帮助,欢迎转载,烦请注明出处,谢谢!

Python 相关文章推荐
Django 生成登陆验证码代码分享
Dec 12 Python
Python程序运行原理图文解析
Feb 10 Python
Python实现获取邮箱内容并解析的方法示例
Jun 16 Python
influx+grafana自定义python采集数据和一些坑的总结
Sep 17 Python
对IPython交互模式下的退出方法详解
Feb 16 Python
使用Python将Mysql的查询数据导出到文件的方法
Feb 25 Python
PyCharm中Matplotlib绘图不能显示UI效果的问题解决
Mar 12 Python
jupyter 导入csv文件方式
Apr 21 Python
Python Selenium截图功能实现代码
Apr 26 Python
在django admin中配置搜索域是一个外键时的处理方法
May 20 Python
基于python实现可视化生成二维码工具
Jul 08 Python
Python字典的基础操作
Nov 01 Python
win10安装tensorflow-gpu1.8.0详细完整步骤
Jan 20 #Python
tensorflow -gpu安装方法(不用自己装cuda,cdnn)
Jan 20 #Python
基于Python获取照片的GPS位置信息
Jan 20 #Python
如何基于pythonnet调用halcon脚本
Jan 20 #Python
使用TensorFlow对图像进行随机旋转的实现示例
Jan 20 #Python
TensorFLow 不同大小图片的TFrecords存取实例
Jan 20 #Python
python各层级目录下import方法代码实例
Jan 20 #Python
You might like
根德YB400的电路分析
2021/03/02 无线电
提问的智慧(2)
2006/10/09 PHP
ThinkPHP使用心得分享-分页类Page的用法
2014/05/15 PHP
Kindeditor编辑器添加图片上传水印功能(php代码)
2017/08/03 PHP
laravel框架中视图的基本使用方法分析
2019/11/23 PHP
javascript下function声明一些小结
2007/12/28 Javascript
对象特征检测法判断浏览器对javascript对象的支持
2009/07/25 Javascript
JS 添加网页桌面快捷方式的代码详细整理
2012/12/27 Javascript
JavaScript解析URL参数示例代码
2013/08/12 Javascript
JavaScript中的对象序列化介绍
2014/12/30 Javascript
JavaScript实现将UPC转换成ISBN的方法
2015/05/26 Javascript
详解Vue自定义过滤器的实现
2017/01/10 Javascript
Angular2平滑升级到Angular4的步骤详解
2017/03/29 Javascript
分析JavaScript数组操作难点
2017/12/18 Javascript
vue项目中跳转到外部链接的实例讲解
2018/09/20 Javascript
vue 实现在同一界面实现组件的动态添加和删除功能
2020/06/16 Javascript
基于vue中的scoped坑点解说
2020/09/04 Javascript
在elementui中Notification组件添加点击事件实例
2020/11/11 Javascript
[01:08:43]DOTA2-DPC中国联赛定级赛 Phoenix vs DLG BO3第一场 1月9日
2021/03/11 DOTA
Python中exit、return、sys.exit()等使用实例和区别
2015/05/28 Python
Python之reload流程实例代码解析
2018/01/29 Python
python jieba分词并统计词频后输出结果到Excel和txt文档方法
2018/02/11 Python
python实现冒泡排序算法的两种方法
2018/03/10 Python
使用Python的datetime库处理时间(RPA流程)
2019/11/24 Python
详解Django中views数据查询使用locals()函数进行优化
2020/08/24 Python
css3新单位vw、vh的使用教程
2018/03/23 HTML / CSS
澳大利亚优质的家居用品和生活方式公司:Bed Bath N’ Table
2019/04/16 全球购物
在数据文件自动增长时,自动增长是否会阻塞对文件的更新
2014/05/01 面试题
short s1 = 1; s1 = s1 + 1;有什么错? short s1 = 1; s1 += 1;有什么错?
2014/09/26 面试题
项目经理岗位职责
2013/11/11 职场文书
幼儿园美术教学反思
2014/01/31 职场文书
驾驶员培训方案
2014/05/01 职场文书
消防宣传口号
2014/06/16 职场文书
新党章心得体会
2014/09/04 职场文书
小学三八妇女节活动总结
2015/02/06 职场文书
基于PyTorch实现一个简单的CNN图像分类器
2021/05/29 Python