常见深度学习模型格式与对应加载执行库

部署运行你感兴趣的模型镜像
模型格式主要来源/用途对应加载与执行库支持语言/平台
PyTorch .pt / .pth / .ptlPyTorch 保存的模型或权重🔸 torch.load()(训练)🔸 torch.jit.load()(TorchScript 推理)Python / C++
TorchScript (.pt)PyTorch 的可执行格式(图模式)libtorchtorch.jitC++、Python
ONNX .onnx通用开放格式,可由 PyTorch、TensorFlow、MXNet 等导出🔸 onnxruntime(微软)🔸 TensorRT(NVIDIA)🔸 OpenVINO(Intel)🔸 MNN(阿里)🔸 TFLite(转换后)多平台(CPU/GPU/NPU)
TensorFlow SavedModel / .pbTensorFlow 模型保存格式🔸 tensorflow(官方)🔸 tf-serving(服务端)🔸 tflite(移动端)Python / C++ / Java
Keras .h5 / .kerasKeras 训练的模型🔸 keras.models.load_model()Python
TensorFlow Lite .tflite移动端 / 边缘设备优化格式🔸 tflite.InterpreterAndroid / iOS / Raspberry Pi
CoreML .mlmodel苹果生态模型格式🔸 coremltools / iOS CoreML RuntimemacOS / iOS
TensorRT .plan / .engineNVIDIA GPU 优化格式🔸 tensorrt Runtime / C++ APILinux / Windows(GPU)
OpenVINO IR .xml + .binIntel 推理优化格式🔸 openvino.runtime.Core()Intel CPU / GPU / VPU
MXNet .params + .jsonApache MXNet 模型🔸 mxnet.model.load_checkpoint()Python / C++
Caffe .prototxt + .caffemodelCaffe 模型🔸 caffe.Net()C++ / Python
PaddlePaddle .pdmodel + .pdparams百度飞桨🔸 paddle.inference.create_predictor()Python / C++
MindSpore .mindir华为 MindSpore🔸 mindspore.load_checkpoint() / load_mindir()Python
gguf / ggml / safetensorsLLM(大语言模型)高效格式🔸 llama.cpptransformersctransformersggmlC++ / Python

您可能感兴趣的与本文相关的镜像

PyTorch 2.5

PyTorch 2.5

PyTorch
Cuda

PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

daolongzhang

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值