|
|
@@ -0,0 +1,12 @@
|
|
|
+#!/bin/sh
|
|
|
+git clone https://github.com/pytorch/serve.git
|
|
|
+cd serve/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/)
|
|
|
+./build_image.sh -g -cv cu111 -t torchserve:1.0 -bt production -b master
|
