RuntimeError: Exporting the operator affine_grid_generator to ONNX opset version 9 is not supported.

最新推荐文章于 2024-03-04 23:43:17 发布
最新推荐文章于 2024-03-04 23:43:17 发布
阅读量929 收藏 6
点赞数 1
分类专栏： Python 目标检测 opencv 文章标签： 深度学习 pytorch 人工智能
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/pyxiaosheng/article/details/127675806
版权
Python 同时被 3 个专栏收录
4 篇文章 0 订阅
订阅专栏
目标检测
4 篇文章 0 订阅
订阅专栏
opencv
2 篇文章 0 订阅
订阅专栏

pth转onnx遇见的坑

RuntimeError: Exporting the operator affine_grid_generator to ONNX opset version 9 is not supported. Please feel free to request support or submit a pull request on PyTorch GitHub.

 解决办法：

原torch.onnx.export

torch.onnx.export(STN, input, onnx_path, verbose=True, input_names=input_names, output_names=output_names) # 指定模型的输入，以及onnx的输出路径

改为：

torch.onnx.export(STN, input, onnx_path, export_params=True, verbose=True,
operator_export_type=torch.onnx.OperatorExportTypes.ONNX_ATEN_FALLBACK)

 

参考： https://www.wangt.cc/2020/10/runtimeerror-exporting-the-operator-max_unpool2d-to-onnx

 

 

                

        

        

    

  


            

                    
            

    

      

        

            

              
                
                  凡人炼丹传
                
              
            

            

                关注
              
            

        

        

          
      

      

            

                专栏目录
          









                



	

		

			

				
					
RuntimeError: CUDA error: CUBLAS_STATUS_INVALID_VALUE when calling `cublasGemmEx 解决方案

				
			

			

				

					weixin_43178406的博客
				

				

					06-12
					
					8万+
					
				

			

		

		

			
				
本文主要介绍了RuntimeError: CUDA error: CUBLAS_STATUS_INVALID_VALUE when calling `cublasGemmEx 解决方案，希望能对学习pytorch的同学们有所帮助。

文章目录

1. 问题描述
2. 解决方案

			
		

	



                

              
                



	

		

			

				
					
RuntimeError: “addmm_impl_cpu_“ not implemented for ‘Half‘解决方案

					
最新发布

				
			

			

				

					weixin_43178406的博客
				

				

					04-18
					
					7万+
					
				

			

		

		

			
				
本文主要介绍了RuntimeError: “addmm_impl_cpu_” not implemented for 'Half’解决方案，希望能对学习大模型的同学们有所帮助。

文章目录

1. 问题描述
2. 解决方案

			
		

	



                


  
              

                


	

		

			

				
					
RuntimeError: Exporting the operator repeat_interleave to ONNX opset version 9 is not supported.

				
			

			

				

					一舫的博客
				

				

					08-27
					
					4687
					
				

			

		

		

			
				
RuntimeError: Exporting the operator repeat_interleave to ONNX opset version 9 is not supported. Please open a bug to request ONNX export support for the missing operator.


pytorch转onnx报的错，原因是模型中的repeat_interleave操作在onnx中还不支持，可以把这个操作换成别的，emmm.....，我不知道换成.

			
		

	





	

		

			

				
					
【mindspore模型复现】【从pytorch复现】torch.nn.functional.affine_grid对应算子缺失

				
			

			

				

					weixin_45666880的博客
				

				

					08-21
					
					132
					
				

			

		

		

			
				
torch.nn.functional.affine_grid在mindspore是不是没有对应算子？暂时还没有torch.nn.functional.affine_grid的对标算子。

			
		

	



		

			
		



	

		

			

				
					
Pytorch中的仿射变换(affine_grid)

				
			

			

				

					bbbeoy的专栏
				

				

					09-01
					
					6100
					
				

			

		

		

			
				
在看 pytorch 的 Spatial Transformer Network 教程 时，在 stn 层中的 affine_grid 与 grid_sample 函数上卡住了，不知道这两个函数该如何使用，经过一些实验终于搞清楚了其作用。


参考：详细解读Spatial Transformer Networks (STN)，该文章与李宏毅的课程一样，推荐听李老师的 STN 这一课，讲的比较清楚；


假设我们有这么一张图片：





魁拔中的卡拉肖克·玲

下面我们将通过分别通过手动编码和pytorc.

			
		

	





	

		

			

				
					
使用Pytorch导出自定义ONNX算子

				
			

			

				

					霹雳吧啦Wz
				

				

					03-04
					
					2143
					
				

			

		

		

			
				
在实际部署模型时有时可能会遇到想用的算子无法导出onnx，但实际部署的框架是支持该算子的。此时可以通过自定义onnx算子的方式导出onnx模型（注：自定义onnx算子导出onnx模型后是无法使用onnxruntime推理的）。但对于当前导出onnx场景可以不用管它，后面的参数是实际自己传入的参数。方法指定具体导出自定义算子的名称，以及输入的参数（注：上面示例中传入的都是。打开自己导出的onnx文件，可以看到如下所示网络结构。，后面的参数任为实际自己传入的参数，然后通过。方法是在导出onnx时调用的函数。

			
		

	





	

		

			

				
					
导出onnx模型，onnx不支持roll操作符

				
			

			

				

					weixin_44791757的博客
				

				

					04-26
					
					3069
					
				

			

		

		

			
				
当我torch.onnx.export（）函数导出swin-tranformer模型时出现报错:



Exporting the operator roll to ONNX opset version 9 is not supported. Please open a bug to request ONNX export support for the missing operator.



解决方法：

将opset version 9换成12（可以先试试只换opset版本，如果其报错不支持这个版本后

			
		

	





	

		

			

				
					
pytorch转onnx: step = 1 is currently not supported以及Exporting the operator silu to ONNX opset version

				
			

			

				

					weixin_37889356的博客
				

				

					12-02
					
					6725
					
				

			

		

		

			
				
在yolov5s的pytorch模型转换onnx模型时报如下错误：


RuntimeError: step!=1 is currently not supported


原因主要是低版本的opset不支持切片操作导致的；

把模型转换的代码改成如下所示即可，即使用版本11以上的opset：


torch.onnx.export(model, img, "xxx.onnx", verbose=True,opset_version=11,export_params=True)

解决该问题后可能会继续出现

			
		

	





	

		

			

				
					
RuntimeError:Internal: src/sentencepiece_processor.cc(1101)model_proto-＞ParseFromArray(serialize解决方案

					
热门推荐

				
			

			

				

					weixin_43178406的博客
				

				

					04-25
					
					9万+
					
				

			

		

		

			
				
本文主要介绍了RuntimeError: Internal: src/sentencepiece_processor.cc(1101) [model_proto->ParseFromArray(serialized.data(), serialized.size())]解决方案，希望对同学们有所帮助。

文章目录

1. 问题描述
2. 解决方案

			
		

	





	

		

			

				
					
RuntimeError: The current installed version of g++ (4.8.5) is less than the mininum version CUDA解决方案

				
			

			

				

					weixin_43178406的博客
				

				

					06-06
					
					8万+
					
				

			

		

		

			
				
本文主要介绍了RuntimeError: The current installed version of g++ (4.8.5) is less than the mininum version CUDA解决方案，希望能对学习python的同学们有所帮助。

文章目录

1. 问题描述
2. 解决方案

			
		

	





	

		

			

				
					
RuntimeError: Exporting the operator grid_sampler to ONNX opset version 11 is not support

				
			

			

				

					欢迎订阅 -- 小源笔迹
				

				

					09-08
					
					2205
					
				

			

		

		

			
				
grid_sampler是比较经典的算子不支持问题，在pytorch1.12之后已经支持了，而我这里要求的版本是torch=1.8.0，该了torch版本后，之后的推理平台又会有版本限制所以不能改。然后用mmcv库中的bilinear_grid_sample()来替换grid_sample函数。二者是等价的，可以去torch官网看具体的api。具体做法就是定位到你出错的代码位置，一般就是grid_sample函数的定义位置。如果还是报错则重新写bilinear_grid_sample函数，具体代码如下。

			
		

	





	

		

			

				
					
[Debug] pytorch导出onnx模型时不支持roll操作

				
			

			

				

					ccheiheihei的博客
				

				

					07-01
					
					1177
					
				

			

		

		

			
				
BUG:RuntimeError: Exporting the operator roll to ONNX opset version 11 is not supported. Please open a bug to request ONNX export support for the missing operator.

			
		

	





	

		

			

				
					
torch转onnx遇到的坑（一）

				
			

			

				

					JoeyChen1219的博客
				

				

					11-10
					
					1万+
					
				

			

		

		

			
				
Pytorch转ONNX遇到的问题及解决方案ONNX不支持torch.linspaceONNX不支持torch中的grid_sampler操作动态输入问题
ONNX不支持torch.linspace
报错提示
RuntimeError: Exporting the operator linspace to ONNX opset version 11 is not supported. Please feel free to request support or submit a pull request o

			
		

	





	

		

			

				
					
onnx踩坑记录

				
			

			

				

					zhang2362167998的博客
				

				

					03-28
					
					1191
					
				

			

		

		

			
				
TypeError: forward() takes 2 positional arguments but 3 were given
ONNX export of operator adaptive pooling, since output_size is not constant
Exporting the operator grid_sampler to ONNX opset version 13 is not supported.
RuntimeError: view size is not co

			
		

	





	

		

			

				
					
.pth转.onnx出现问题解决。（记录，仅供参考）

				
			

			

				

					sadjhaksdas的博客
				

				

					03-07
					
					1774
					
				

			

		

		

			
				
pth权重模型转换onnx

			
		

	





	

		

			

				
					
利用pytorch的affine_grid和grid_sample实现rroi_align

				
			

			

				

					怡宝2号
				

				

					06-22
					
					4018
					
				

			

		

		

			
				
原始图片：

import random
import math
import torch
import numpy as np
import torch.nn.functional as F
import cv2
import matplotlib.pyplot as plt
from data_gen import draw_box_points

path = './test/timg.jp...

			
		

	





	

		

			

				
					
RuntimeError: Exporting the operator uniform to ONNX opset version 12 is not supported.

				
			

			

				

					我的小天地 ^_^
				

				

					01-10
					
					4568
					
				

			

		

		

			
				
pt转onnx报错：
RuntimeError: Exporting the operator uniform to ONNX opset version 12 is not supported. 
Please open a bug to request ONNX export support for the missing operator.

原因考虑是转换的时候，不可以在forward中动态获取值再进行运算，而应该在init中就完成相应的操作。
不便上传图片，有问题留言。

...

			
		

	





	

		

			

				
					
第二步：pytorch模型转onnx模型步骤和可能遇到的问题

				
			

			

				

					yx903520的博客
				

				

					05-25
					
					1万+
					
				

			

		

		

			
				
一、本地(win10)环境
cuda11.0 + python3.7.4 + pytorch1.7.1 + onnx1.9.0
二、可能出现的问题
1. pytorch与tensorrt版本对应问题
pytorch转onnx模型是为了后续导入tensorrt做准备，所以要考虑pytorch与tensorrt版本对应的问题。由于服务器的环境是cuda10.1 + tensorrt7.0.0.11，所以要先把pytorch模型从1.7.1降到1.4.0。以efficientnet_b1为例，先在pytorch1

			
		

	





	

		

			

				
					
pytorch affine_grid

				
			

			

				

					u012796629的博客
				

				

					08-10
					
					562
					
				

			

		

		

			
				
theta: 0:tx 1:ty

grid 0:x(width) 1:y(height)


			
		

	





	

		

			

				
					
RuntimeError: CUDA error: CUBLAS_STATUS_NOT_SUPPORTED when calling `cublasSgemm

				
			

			

				

					08-17
				

			

		

		

			
				
根据引用中的信息，您遇到的错误是"RuntimeError: CUDA error: CUBLAS_STATUS_NOT_SUPPORTED when calling `cublasSgemm`"。这个错误通常是由于CUDA库的不兼容性或配置问题引起的。  其中有一种可能的解决方法是检查...

			
		

	



              




          




        



    





    



      

      

        
      

      





	

		
评论 4

	

	
	




  

    

      添加红包
      
    

    

      

        
        

          
          
        

        
请填写红包祝福语或标题

      

      

        
        

          
          个
        

        
红包个数最小为10个

      

      

        
        

          
          元
        

        
红包金额最低5元

      

      

        
        

          当前余额3.43元
          前往充值 >
        

      

      

        

          需支付：10.00元
        

        
        
      

    

  





  

    
    
  

  

    
    
  








  

    

      

        

          

            
            
成就一亿技术人!

          

          

        

        

          

          

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

        

      

      

        

          

            
              
            
            

              
hope_wisdom
 发出的红包
            

          

        

        

          

          

          

        

      

    

    

  



      
      

      
实付元

      
使用余额支付

      

        

          

            
            点击重新获取
          

        

        
扫码支付

      

      

        
          
            
          
          
        
      

      

        
        钱包余额
          0
          

            
            

              

                
抵扣说明：

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

              

            

          

      

      余额充值