123456789101112131415161718192021222324 |
- #!/bin/sh
- {
- git clone https://github.com/pytorch/serve.git
- # if git clone takes to long time, you should change etc/hosts refer to https://blog.csdn.net/qq_47911262/article/details/121918174:
- } || {
- echo "git clone https://github.com/pytorch/serve.git failed 🤢🤢🤢"
- }
- echo "git clone https://github.com/pytorch/serve.git successfully 😊😊😊"
- cd serve/docker &&
- echo "cd server/docker..." &&
- # -h, --help Show script help
- # -b, --branch_name Specify a branch name to use. Default: master
- # -g, --gpu Build image with GPU based ubuntu base image
- # -bt, --buildtype Which type of docker image to build. Can be one of : production(default), dev, codebuild
- # -t, --tag Tag name for image. If not specified, script uses torchserve default tag names.
- # -cv, --cudaversion Specify to cuda version to use. Supported values cu92, cu101, cu102, cu111, cu113. Default cu102
- # --codebuild Set if you need AWS CodeBuild(https://aws.amazon.com/cn/codebuild/)
- echo "./build_image.sh -g -cv cu111 -t torchserve:cuda11-1.0 -bt production -b master ..." &&
- ./build_image.sh -g -cv cu111 -t torchserve:1.0 -bt production -b master &&
- echo "install torcher_server docker finally ~" &&
- pip3 install torchserve -i https://pypi.tuna.tsinghua.edu.cn/simple &&
- pip3 install torch-model-archiver -i https://pypi.tuna.tsinghua.edu.cn/simple &&
- pip3 install torch-workflow-archiver -i https://pypi.tuna.tsinghua.edu.cn/simple &&
- pip3 install chardet -i https://pypi.tuna.tsinghua.edu.cn/simple
|