ubntu16.04 tensorflow模型ckpt转pb时遇到的问题

最新推荐文章于 2022-11-03 11:53:21 发布
最新推荐文章于 2022-11-03 11:53:21 发布
阅读量2k 收藏 3
点赞数
分类专栏： AI 文章标签： ubuntu tensorflow CUDA nvidia linux
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/m2008h1/article/details/84786977
版权
在Ubuntu16.04上，将TensorFlow的ckpt模型转换为pb格式时遇到错误，包括'IteratorGetDevice' KeyError、'NcclAllReduce' OpKernel注册错误、cudaGetDevice失败以及参数值加载问题。错误源于模型构建时使用了tf.contrib，不同TF版本间的不兼容，以及旧版NVIDIA驱动。解决方案涉及匹配TF版本、更新NVIDIA驱动和处理变量类型转换。
摘要由CSDN通过智能技术生成

问题1：在冻结一个ckpt格式的模型为pb格式时，需要加载图数据，使用

tf.train.import_meta_graph

遇到的报错如下：

op_def = op_dict[node.op]
KeyError: 'IteratorGetDevice'

冻结的代码是没有问题，另一个模型是可以正常转换为pb格式的。经过网络查询，可以确定语音是模型构建时使用了tf.contrib的内容，大概是转换时的一些兼容问题。

截图源自：https://www.bountysource.com/issues/61636124-keyerror-iteratorgetdevice-when-using-official-pre-trained-resnet-model

解决方案：


                

        

        

    

  


        

            

                    
                      最低0.47元/天 解锁文章
                        
                    
            

        

    

      

        

            

              
                
                  m2008h1
                
              
            

            

                关注
              
            

        

        

          
      

      

            

                专栏目录
          









                



	

		

			

				
					
报错：cudaGetDevice() failed. Status: CUDA driver version is insufficient for CUDA runtime version

				
			

			

				

					u010513327的博客
				

				

					03-03
					
					2万+
					
				

			

		

		

			
				
在GPU上运行TensorFlow程序时报错：cudaGetDevice() failed. Status: CUDA driver version is insufficient for CUDA runtime version

可能是cuda和显卡驱动不一致

CUDA版本与显卡驱动版本匹配查询：https://docs.nvidia.com/cuda/cuda-toolkit-releas...

			
		

	



                

              
                



	

		

			

				
					
TensorFlow把ckpt转化为pb文件代码

				
			

			

				

					12-13
				

			

		

		

			
				
把tensorflow保存的checkpoint类型模型冻结，转化为.pb模型输出。调用方法参见我的博文https://blog.csdn.net/weixin_41864878/article/details/84957681

			
		

	



                


  
              

                


	

		

			

				
					
TensorFlow 自定义模型导出：将 .ckpt 格式转化为 .pb 格式

				
			

			

				

					
				

				

					08-22
					
					715
					
				

			

		

		

			
				
本文承接上文 TensorFlow-slim 训练 CNN 分类模型（续），阐述通过 tf.contrib.slim 的函数 slim.learning.train 训练的模型，怎么通过人为的加入数据入口（即占位符）来克服无法用于图像推断的问题。要解决这个问题，最简单和最省时的方法是模仿。我们模仿的代码是 TensorFlow 实现的目标检测 API 中的文件 exporter...

			
		

	





	

		

			

				
					
tensorflow的ckpt文件转pb模型文件

				
			

			

				

					10-28
				

			

		

		

			
				
使用这个python程序可以直接将ckpt文件转成固化的模型文件，用于预测结果，注意修改路径

			
		

	



		

			
		



	

		

			

				
					
ImageProjectiveTransform 错误解决办法

				
			

			

				

					Peacemaker的博客
				

				

					01-06
					
					1324
					
				

			

		

		

			
				
 

tf.train.import_meta_graph  报错


Traceback (most recent call last):
  File "freeze_graph.py", line 109, in <module>
    main(parse_arguments(sys.argv[1:]))
  File "freeze_graph.py", line 50, ...

			
		

	





	

		

			

				
					
打造微信小程序之人脸属性（ubuntu16.04+tensorflow）

				
			

			

				

					li-shaoyu的博客
				

				

					06-11
					
					483
					
				

			

		

		

			
				
最近学习小程序开发，涉及到了下列内容：

1.数据打包


##creat_data.py
##实现数据的打包
import cv2
import tensorflow as tf
##dlib  实现抠图
import dlib

##读取标志信息索引值,不同的索引值对应不同的属性
anno_file = "/home/lsy/CelebA/Anno/list_attr_celeba.txt"
ff = open(anno_file)

anno_info = ff.readlines()

attrib

			
		

	





	

		

			

				
					
Ubuntu16.04 配置TensorFlow 1.10.1（Object Detection API）运行MobileNet-SSD（ssd_mobilenet_v1_coco）

				
			

			

				

					Arvin_liang的博客
				

				

					11-28
					
					4548
					
				

			

		

		

			
				
Ubuntu16.04 配置TensorFlow 1.10.1（Object Detection API）运行MobileNet-SSD（ssd_mobilenet_v1_coco）TensorFlow的安装方式从源码安装1. 克隆 TensorFlow 仓库2. 安装 Bazel及其他依赖  1. 下载二进制安装程序  2. 安装所需包  3. 运行安装程序  4. 设置环境变量3. 创建 pi...

			
		

	





	

		

			

				
					
ubuntu16.04+python+anaconda+tensorflow的深度学习2

				
			

			

				

					guosuling的博客
				

				

					01-13
					
					945
					
				

			

		

		

			
				
目录



1、Tensorflow源代码解析

1.1 TensorFlow 的目录结构

1.1.1 contirb

1.1.2 core

1.1.3 examples

1.1.4python

1.2 TensorFlow 源代码的学习方法

2、机器学习基础

2.1 有监督学习简介

拓展知识1  协调器 tf.train.Coordinator 和入队线程启动器...

			
		

	





	

		

			

				
					
基于TensorFlow Object Detection API使用ssd_mobilenet_v1训练自己的数据集并转换为tflite模型（亲测可用）

				
			

			

				

					zjj7725的博客
				

				

					12-03
					
					2944
					
				

			

		

		

			
				
使用TensorFlow Object Detection API中mobilenet_ssd_v1训练并转换为tflite模型
1. 环境
OS: Ubuntu 16.04 x64
Anaconda: 4.6.12
Python: 3.6.8
TensorFlow（GPU版）: 1.13.1
OpenCV: 3.4.1
2. 基础环境配置
Anaconda 下载地址: Anaconda-4.6....

			
		

	





	

		

			

				
					
报错tensorflow.python.framework.errors_impl.InternalError，from device: CUDA_ERROR_OUT_OF_MEMORY

				
			

			

				

					Hello World! Keep Coding
				

				

					11-23
					
					3755
					
				

			

		

		

			
				
此文为解决机器学习中使用tensorflow，在运行代码出现上述报错情况

1.先运行nvidia-smi      检查GPU运行情况，若内存够用进入2

2.代码应作已下修改

import tensorflow as tf
import os
os.environ["CUDA_VISIBLE_DEVICES"] = '0' #use GPU with ID=0
config = tf.Con...

			
		

	





	

		

			

				
					
InternalError: cudaGetDevice() failed. Status: CUDA driver version is insufficient for CUDA runtime

					
最新发布

				
			

			

				

					mynameisyaxuan的博客
				

				

					11-03
					
					493
					
				

			

		

		

			
				
InternalError: cudaGetDevice() failed. Status: CUDA driver version is insufficient for CUDA runtime version 

			
		

	





	

		

			

				
					
【Tensorflow】No OpKernel was registered to support Op ‘NcclAllReduce‘错误

				
			

			

				

					豆芽菜
				

				

					04-30
					
					1772
					
				

			

		

		

			
				
问题：

使用tf.distribute.MirroredStrategy时，在windows系统多卡下出现No OpKernel was registered to support Op 'NcclAllReduce'问题（Linux下正常），具体报错信息如下：


tensorflow.python.framework.errors_impl.InvalidArgumentError: No OpKernel was registered to support Op 'NcclAllRe...

			
		

	





	

		

			

				
					
PB中，将内容输出到Word书签时遇到的问题

				
			

			

				

					weixin_30716141的博客
				

				

					09-10
					
					236
					
				

			

		

		

			
				
代码：  ole_object.selection.goto(true,0,0,'P11')   ole_object.activedocument.bookmarks.item('P11').select    ole_object.selection.typetext(down_g2) //G2       原因：typetext()函数中的参数需是字符串变量 在这里可以用String(do...

			
		

	





	

		

			

				
					
RTX3080无法运行TensorFlow的若干问题

				
			

			

				

					dcct1990的博客
				

				

					02-20
					
					2166
					
				

			

		

		

			
				
RTX3080 TensorFlow报错

			
		

	





	

		

			

				
					
Ubuntu 16.04配置TensorFlow-GPU1.6.0全攻略

				
			

			

				

					
				

			

		

		

			
				
"Ubuntu 16.04×86_64系统下配置TensorFlow-GPU 1.6.0的详细步骤，包括环境检查、CUDA 9.0与cuDNN 7.0的安装、环境变量配置及TensorFlow-GPU的安装与测试。"  在Ubuntu 16.04×86_64系统上配置TensorFlow-GPU 1.6.0...

			
		

	



              




          




        



    





    



      

      

        
      

      





	

		
评论 1

	

	
	




  

    

      添加红包
      
    

    

      

        
        

          
          
        

        
请填写红包祝福语或标题

      

      

        
        

          
          个
        

        
红包个数最小为10个

      

      

        
        

          
          元
        

        
红包金额最低5元

      

      

        
        

          当前余额3.43元
          前往充值 >
        

      

      

        

          需支付：10.00元
        

        
        
      

    

  





  

    
    
  

  

    
    
  








  

    

      

        

          

            
            
成就一亿技术人!

          

          

        

        

          

          

            领取后你会自动成为博主和红包主的粉丝
            规则
          

        

      

      

        

          

            
              
            
            

              
hope_wisdom
 发出的红包
            

          

        

        

          

          

          

        

      

    

    

  



      
      

      
实付元

      
使用余额支付

      

        

          

            
            点击重新获取
          

        

        
扫码支付

      

      

        
          
            
          
          
        
      

      

        
        钱包余额
          0
          

            
            

              

                
抵扣说明：

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

              

            

          

      

      余额充值