Skip to content

study-eq-eat-drink/model-serving

Repository files navigation

model-serving

모델 서빙에 가장 적합한 도구를 찾자

테스트 조합

  1. triton + tensorflow
  2. triton + tensorflow-onnx
  3. triton + tensorflow-onnx-tensorRT
  4. bentoML + tensorflow
  5. bentoML + tensorflow-onnx
  6. akka + tensorflow
  7. akka + tensorflow-onnx

Triton build & run

Build

docker build -t benjamin/nsmc/triton-onnx:0.0.1 -f triton-serving/onnx/Dockerfile .
docker build -t benjamin/nsmc/triton-onnx-tensorrt:0.0.1 -f triton-serving/onnx-tensorrt/Dockerfile .

Run

docker run --gpus=all -p 9000:8000 -p 9001:8001 -p 9002:8002 --rm benjamin/nsmc/triton-onnx:0.0.1 tritonserver --model-repository=/models
docker run --gpus=all -p 9000:8000 -p 9001:8001 -p 9002:8002 --rm benjamin/nsmc/triton-onnx-tensorrt:0.0.1 tritonserver --model-repository=/models

About

모델 추론에 가장 적합한 도구를 찾자

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published