mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
30 lines
826 B
Bash
30 lines
826 B
Bash
|
#!/bin/sh
|
||
|
|
|
||
|
|
set -e
|
||
|
|
|
||
|
|
ip=${1}
|
||
|
|
port=${2:8585}
|
||
|
|
|
||
|
|
date_today=`date +'%Y%m%d'`
|
||
|
|
|
||
|
|
# create http server
|
||
|
|
nohup python3 -m http.server --directory /data2/shared/mmdeploy-manylinux2014_x86_64-cuda11.3 $port > tmp.log 2>&1
|
||
|
|
|
||
|
|
export ip=10.1.52.36
|
||
|
|
export port=8585
|
||
|
|
export CUDA_URL=http://$ip:$port/cuda_11.3.0_465.19.01_linux.run
|
||
|
|
export CUDNN_URL=http://$ip:$port/cudnn-11.3-linux-x64-v8.2.1.32.tgz
|
||
|
|
export TENSORRT_URL=http://$ip:$port/TensorRT-8.2.3.0.Linux.x86_64-gnu.cuda-11.4.cudnn8.2.tar.gz
|
||
|
|
export TENSORRT_VERSION=8.2.3.0
|
||
|
|
export TAG=manylinux2014_x86_64-cuda11.3
|
||
|
|
|
||
|
|
# build docker image
|
||
|
|
docker build ./docker/prebuild/ -t openmmlab/mmdeploy:$TAG \
|
||
|
|
--build-arg CUDA_URL=$CUDA_URL \
|
||
|
|
--build-arg CUDNN_URL=$CUDNN_URL \
|
||
|
|
--build-arg TENSORRT_URL=${TENSORRT_URL}
|
||
|
|
|
||
|
|
# push to docker hub
|
||
|
|
docker login
|
||
|
|
docker push openmmlab/mmdeploy:$TAG
|