mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2026-05-10 17:41:13 +08:00
Fix serving/scripts build.sh (#1053)
* fix serving/scripts build.sh * Update build.sh support: ``` bash build.sh --docker_name=test ``` * Update build.sh Co-authored-by: heliqi <1101791222@qq.com>
This commit is contained in:
@@ -12,7 +12,41 @@
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
WITH_GPU=${1:-ON}
|
||||
|
||||
ARGS=`getopt -a -o w:n:h:hs -l WITH_GPU:,docker_name:,http_proxy:,https_proxy: -- "$@"`
|
||||
|
||||
eval set -- "${ARGS}"
|
||||
echo "parse start"
|
||||
|
||||
while true
|
||||
do
|
||||
case "$1" in
|
||||
-w|--WITH_GPU)
|
||||
WITH_GPU="$2"
|
||||
shift;;
|
||||
-n|--docker_name)
|
||||
docker_name="$2"
|
||||
shift;;
|
||||
-h|--http_proxy)
|
||||
http_proxy="$2"
|
||||
shift;;
|
||||
-hs|--https_proxy)
|
||||
https_proxy="$2"
|
||||
shift;;
|
||||
--)
|
||||
shift
|
||||
break;;
|
||||
esac
|
||||
shift
|
||||
done
|
||||
|
||||
if [ -z $WITH_GPU ];then
|
||||
WITH_GPU="ON"
|
||||
fi
|
||||
|
||||
if [ -z $docker_name ];then
|
||||
docker_name="build_fd"
|
||||
fi
|
||||
|
||||
if [ $WITH_GPU == "ON" ]; then
|
||||
|
||||
@@ -30,7 +64,7 @@ if [ ! -d "./TensorRT-8.4.1.5/" ]; then
|
||||
rm -rf TensorRT-8.4.1.5.Linux.x86_64-gnu.cuda-11.6.cudnn8.4.tar.gz
|
||||
fi
|
||||
|
||||
nvidia-docker run -i --rm --name build_fd \
|
||||
nvidia-docker run -i --rm --name ${docker_name} \
|
||||
-v`pwd`/..:/workspace/fastdeploy \
|
||||
-e "http_proxy=${http_proxy}" \
|
||||
-e "https_proxy=${https_proxy}" \
|
||||
@@ -68,7 +102,7 @@ else
|
||||
|
||||
echo "start build FD CPU library"
|
||||
|
||||
docker run -i --rm --name build_fd \
|
||||
docker run -i --rm --name ${docker_name} \
|
||||
-v`pwd`/..:/workspace/fastdeploy \
|
||||
-e "http_proxy=${http_proxy}" \
|
||||
-e "https_proxy=${https_proxy}" \
|
||||
|
||||
Reference in New Issue
Block a user