mirror of
https://github.com/hanzo-ml/kserve.git
synced 2026-10-30 20:53:49 +00:00
jagadeeshi2iandGitHub
348cc85f53
* Fix: Change folder structure - move torchserve from custom to v1beta1 -> custom * Fix: Update readme for samples * Fix: Update readme with links and description for torchserve
723 B
723 B
Docker image building
Steps:
- Download model archive file from the model-zoo or create you own using the step provided here
- Copy model archive files to model-store folder
- Edit config.properties for requirement
- Run docker build
- Publish the image to dockerhub repo
# For CPU:
DOCKER_BUILDKIT=1 docker build --file Dockerfile -t torchserve:latest .
# For GPU:
DOCKER_BUILDKIT=1 docker build --file Dockerfile --build-arg BASE_IMAGE=nvidia/cuda:10.1-cudnn7-runtime-ubuntu18.04 -t torchserve-gpu:latest .
docker push {username}/torchserve:latest