server_develop/src_code/build_docker_container.sh

#!/bin/sh
docker run \
      --rm \
      -it  \
      --gpus '"device=0"' \
      -p 8080:8080        \
      -p 8081:8081        \
      -p 8082:8082        \
      -p 7070:7070        \
      -p 7071:7071        \
      --name yolov5_torch_serve-gpu    \
      -e install_py_dep_per_model=true \
      --mount type=bind,source=/$(pwd)/models-server/model-store,target=/home/models-server/model-store \
      torchserve:cu11-1.0