引擎配置
包括PaddlePaddle, Pytorch, MxNet, Tensorflow使用CPU计算的配置(自动配置,手动配置)。
初次运行时,自动配置会从亚马逊服务器下载native引擎,所以速度可能会比较慢。
而手动配置会从maven服务器(可以指向国内服务器)下载,速度会很快。并且本地库会一起打包,部署后运行时,不会再加载本地运行库。
架构:
支持的引擎及操作系统:
- MXNet - full support
- PyTorch - full support
- TensorFlow - supports inference and NDArray operations
- ONNX Runtime - supports basic inference
- PaddlePaddle - supports basic inference
- TFLite - supports basic inference
- TensorRT - supports basic inference
- DLR - supports basic inference
详细的配置信息(包括CPU,GPU,自动配置,离线配置)参考链接:
离线配置参考例子
下面是配置参考例子:
version 0.20.0(版本可能不会及时维护,请参考上面的文档)
1. PaddlePaddle engine
1). 自动配置
2). macOS - CPU
3). Linux - CPU
4). Linux - GPU
需要设置环境变量LD_LIBRARY_PATH,如在 /etc/profile中设置。
- CUDA 支持的版本
5). Windows - CPU
6). Windows GPU
- CUDA 支持的版本
2. Pytorch engine
1). 自动配置
如果需要明确的指定pytorch版本,可以用下面的配置方法:
|
PyTorch engine version
|
PyTorch native library version
|
|---|---|
|
pytorch-engine:0.20.0
|
1.11.0, 1.12.1, 1.13.0
|
|
pytorch-engine:0.19.0
|
1.10.0, 1.11.0, 1.12.1
|
|
pytorch-engine:0.18.0
|
1.9.1, 1.10.0, 1.11.0
|
|
pytorch-engine:0.17.0
|
1.9.1, 1.10.0, 1.11.0
|
2). macOS - CPU
3). macOS - M1
4). Linux - CPU
5). For aarch64 build
6). For Pre-CXX11 build
7). Linux - GPU
- CUDA 支持的版本
8). Windows - CPU
9). Windows - GPU
- CUDA 支持的版本
3. MxNet engine
1). 自动配置
2). macOS - CPU
3). Linux - CPU
4). Linux - GPU
- CUDA 支持的版本
5). Windows - CPU
6). Windows - GPU
- CUDA 支持的版本
4. Tensorflow engine
1). 自动配置
2). macOS - CPU
3). Linux - CPU
4). Linux - GPU
- CUDA 支持的版本
5). Windows - CPU
6). Windows - GPU
- CUDA 支持的版本