- 博客(7)
- 资源 (1)
- 收藏
- 关注
原创 torch转onnx遇到的坑(一)
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
2021-11-10 14:21:30 15212 50
原创 MobaXterm使用与Anaconda环境配置
MobaXterm的安装使用及 linux服务器端的anaconda安装使用 torch安装及使用 tensorboard的本地访问
2021-10-22 17:26:15 4916
原创 Could not load library libcudnn_ops_infer.so.8.
看了很多网上的方案都无法解决,尝试了找到环境中libcudnn_ops_infer.so.8文件的位置,并将其加入 ~/.bashrc 环境配置文件中,依然提示报错。如果你也是在之后突然无法运行之前的代码,不妨试试我的解决方案。
2024-09-14 17:46:01 320
原创 PyTorch模型转 apple CoreML模型
如果要部署在苹果mac端,官方已经不推荐使用onnx模型转换CoreML模型了,现在最新支持torch模型直接通过cmtools转换到CoreML模型。
2024-04-23 17:03:43 510
原创 PyTorch模型转onnx模型
onnx模型是多家巨头公司共同维护的中间模型表示,很多端侧推理框架的部署都需要用onnx作为中间模型转换到相应推理框架需求的模型。
2024-04-23 16:25:07 188
原创 torch转onnx遇到的坑(二)
近期在onnxruntime推理过程中遇到了mmcv中grid_sample不支持动态输入的问题,用onnxsim解决了该问题
2022-08-17 15:08:29 2927 3
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人