-
Notifications
You must be signed in to change notification settings - Fork 1
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
52ab2d3
commit f056b9b
Showing
1 changed file
with
214 additions
and
104 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,125 +1,235 @@ | ||
| #!/bin/bash | ||
|
|
||
| # run_dockerized.sh | ||
| # A script to run the yolov8_detection Docker container with necessary checks. | ||
| # Function to print error messages in red | ||
| function echo_error { | ||
| echo -e "\e[31mERROR: $1\e[0m" | ||
| } | ||
|
|
||
| # Exit immediately if a command exits with a non-zero status. | ||
| set -e | ||
| # Function to print success messages in green | ||
| function echo_success { | ||
| echo -e "\e[32mSUCCESS: $1\e[0m" | ||
| } | ||
|
|
||
| # Function to display error messages | ||
| error_exit() { | ||
| echo -e "\\033[0;31mERROR:\\033[0m $1" 1>&2 | ||
| exit 1 | ||
| # Function to check if Docker is installed | ||
| function check_docker_installed { | ||
| if ! command -v docker &> /dev/null; then | ||
| echo_error "Docker is not installed. Please install Docker and try again." | ||
| exit 1 | ||
| fi | ||
| } | ||
|
|
||
| # Function to display info messages | ||
| info() { | ||
| echo -e "\\033[0;32mINFO:\\033[0m $1" | ||
| # Function to check if Docker service is running | ||
| function check_docker_service { | ||
| if ! systemctl is-active --quiet docker; then | ||
| echo_error "Docker service is not running. Attempting to start Docker..." | ||
| sudo systemctl start docker | ||
| if ! systemctl is-active --quiet docker; then | ||
| echo_error "Failed to start Docker service. Please start it manually and try again." | ||
| exit 1 | ||
| else | ||
| echo_success "Docker service started successfully." | ||
| fi | ||
| else | ||
| echo_success "Docker service is running." | ||
| fi | ||
| } | ||
|
|
||
| # Function to display warning messages | ||
| warning() { | ||
| echo -e "\\033[0;33mWARNING:\\033[0m $1" | ||
| # Function to check if user is in the docker group | ||
| function check_docker_group { | ||
| if groups "$USER" | grep &>/dev/null '\bdocker\b'; then | ||
| echo_success "User '$USER' is in the 'docker' group. Docker commands can be run without sudo." | ||
| USE_SUDO=false | ||
| else | ||
| echo_error "User '$USER' is NOT in the 'docker' group. Docker commands will require sudo." | ||
| USE_SUDO=true | ||
| fi | ||
| } | ||
|
|
||
| # Check if Docker is installed | ||
| if ! command -v docker &> /dev/null; then | ||
| error_exit "Docker is not installed. Please install Docker and try again. | ||
| You can install Docker by following the instructions at https://docs.docker.com/get-docker/" | ||
| fi | ||
| # Function to check if the Docker image exists | ||
| function check_docker_image { | ||
| if docker image inspect yolov8_detection:latest > /dev/null 2>&1; then | ||
| echo_success "Docker image 'yolov8_detection:latest' is available." | ||
| IMAGE_AVAILABLE=true | ||
| else | ||
| echo_error "Docker image 'yolov8_detection:latest' not found." | ||
| IMAGE_AVAILABLE=false | ||
| fi | ||
| } | ||
|
|
||
| info "Docker is installed." | ||
| # Function to build the Docker image | ||
| function build_docker_image { | ||
| echo "Building Docker image 'yolov8_detection:latest'..." | ||
| if [ "$USE_SUDO" = true ]; then | ||
| sudo docker build -t yolov8_detection:latest . | ||
| else | ||
| docker build -t yolov8_detection:latest . | ||
| fi | ||
|
|
||
| # Check if Docker daemon is running | ||
| if ! sudo docker info > /dev/null 2>&1; then | ||
| error_exit "Docker daemon is not running. Please start Docker and try again. | ||
| You can start Docker with: sudo systemctl start docker" | ||
| fi | ||
| if [ $? -eq 0 ]; then | ||
| echo_success "Docker image 'yolov8_detection:latest' built successfully." | ||
| else | ||
| echo_error "Failed to build Docker image 'yolov8_detection:latest'. Check the Dockerfile and build logs for errors." | ||
| exit 1 | ||
| fi | ||
| } | ||
|
|
||
| info "Docker daemon is running." | ||
|
|
||
| # Check if the user can run Docker without sudo | ||
| if docker ps > /dev/null 2>&1; then | ||
| DOCKER_CMD="docker" | ||
| else | ||
| warning "Current user cannot run Docker commands without sudo." | ||
| read -p "Do you want to run the Docker command with sudo? [y/N]: " USE_SUDO | ||
| case "$USE_SUDO" in | ||
| [yY][eE][sS]|[yY]) DOCKER_CMD="sudo docker" ;; | ||
| *) error_exit "Docker command requires sudo. Exiting." ;; | ||
| esac | ||
| fi | ||
| # Function to run the Docker container | ||
| function run_docker_container { | ||
| echo "Running Docker container 'yolov8_detection:latest'..." | ||
| if [ "$USE_SUDO" = true ]; then | ||
| sudo docker run --gpus all --network=host --rm yolov8_detection:latest | ||
| else | ||
| docker run --gpus all --network=host --rm yolov8_detection:latest | ||
| fi | ||
| } | ||
|
|
||
| # Check if the Docker image exists locally | ||
| IMAGE_NAME="yolov8_detection:latest" | ||
| if ! $DOCKER_CMD image inspect "$IMAGE_NAME" > /dev/null 2>&1; then | ||
| warning "Docker image '$IMAGE_NAME' not found locally." | ||
| read -p "Do you want to build the Docker image now? [Y/n]: " BUILD_IMG | ||
| BUILD_IMG=${BUILD_IMG:-Y} # Default to 'Y' if empty | ||
| case "$BUILD_IMG" in | ||
| [yY][eE][sS]|[yY]|"") | ||
| info "Building Docker image '$IMAGE_NAME'..." | ||
| $DOCKER_CMD build -t "$IMAGE_NAME" . | ||
| info "Docker image '$IMAGE_NAME' built successfully." | ||
| # Main script execution | ||
| check_docker_installed | ||
| check_docker_service | ||
| check_docker_group | ||
| check_docker_image | ||
|
|
||
| if [ "$IMAGE_AVAILABLE" = false ]; then | ||
| echo "Do you want to build the Docker image now? [y/N]" | ||
| read -r BUILD_CHOICE | ||
| case "$BUILD_CHOICE" in | ||
| [yY][eE][sS]|[yY]) | ||
| build_docker_image | ||
| ;; | ||
| *) | ||
| error_exit "Docker image '$IMAGE_NAME' is required. Exiting." | ||
| echo_error "Docker image 'yolov8_detection:latest' is required to run the container. Exiting." | ||
| exit 1 | ||
| ;; | ||
| esac | ||
| else | ||
| info "Docker image '$IMAGE_NAME' is available locally." | ||
| fi | ||
|
|
||
| # Check if the script is running on Linux (since --network=host is Linux-only) | ||
| OS_NAME=$(uname) | ||
| if [ "$OS_NAME" != "Linux" ]; then | ||
| warning "--network=host is only supported on Linux. Adjusting network settings accordingly." | ||
| # Optionally, you can set a different network or notify the user to modify the script | ||
| # For simplicity, we'll proceed without --network=host | ||
| NETWORK_OPTION="" | ||
| else | ||
| NETWORK_OPTION="--network=host" | ||
| fi | ||
|
|
||
| # Check if GPUs are available (optional, requires nvidia-docker) | ||
| if command -v nvidia-smi &> /dev/null; then | ||
| info "NVIDIA GPUs detected." | ||
| GPU_OPTION="--gpus all" | ||
| else | ||
| warning "No NVIDIA GPUs detected or NVIDIA drivers not installed." | ||
| GPU_OPTION="" | ||
| fi | ||
|
|
||
| # Optional: Check if Nginx RTMP server is accessible | ||
| # Uncomment the following block if you want to perform this check | ||
| : <<'END' | ||
| NGINX_URL="rtmp://127.0.0.1:1935/live/stream" | ||
| # Function to check if the RTMP server is reachable | ||
| check_nginx_rtmp() { | ||
| # Attempt to establish a TCP connection to the RTMP server | ||
| nc -zv 127.0.0.1 1935 &> /dev/null | ||
| if [ $? -ne 0 ]; then | ||
| warning "Nginx RTMP server is not accessible at $NGINX_URL." | ||
| read -p "Do you want to proceed anyway? [y/N]: " PROCEED | ||
| case "$PROCEED" in | ||
| [yY][eE][sS]|[yY]) ;; | ||
| *) error_exit "Nginx RTMP server is required. Exiting." ;; | ||
| esac | ||
| else | ||
| info "Nginx RTMP server is accessible at $NGINX_URL." | ||
| fi | ||
| } | ||
| check_nginx_rtmp | ||
| END | ||
|
|
||
| # Run the Docker container | ||
| info "Starting Docker container '$IMAGE_NAME'..." | ||
| $DOCKER_CMD run $GPU_OPTION $NETWORK_OPTION | ||
|
|
||
| # run with: | ||
| # $DOCKER_CMD run $GPU_OPTION $NETWORK_OPTION --rm "$IMAGE_NAME" | ||
| # if you want to delete after running | ||
|
|
||
| info "Docker container '$IMAGE_NAME' has stopped." | ||
| run_docker_container | ||
|
|
||
| # ------------ | ||
| # // (old) | ||
| # #!/bin/bash | ||
|
|
||
| # # run_dockerized.sh | ||
| # # A script to run the yolov8_detection Docker container with necessary checks. | ||
|
|
||
| # # Exit immediately if a command exits with a non-zero status. | ||
| # set -e | ||
|
|
||
| # # Function to display error messages | ||
| # error_exit() { | ||
| # echo -e "\\033[0;31mERROR:\\033[0m $1" 1>&2 | ||
| # exit 1 | ||
| # } | ||
|
|
||
| # # Function to display info messages | ||
| # info() { | ||
| # echo -e "\\033[0;32mINFO:\\033[0m $1" | ||
| # } | ||
|
|
||
| # # Function to display warning messages | ||
| # warning() { | ||
| # echo -e "\\033[0;33mWARNING:\\033[0m $1" | ||
| # } | ||
|
|
||
| # # Check if Docker is installed | ||
| # if ! command -v docker &> /dev/null; then | ||
| # error_exit "Docker is not installed. Please install Docker and try again. | ||
| # You can install Docker by following the instructions at https://docs.docker.com/get-docker/" | ||
| # fi | ||
|
|
||
| # info "Docker is installed." | ||
|
|
||
| # # Check if Docker daemon is running | ||
| # if ! sudo docker info > /dev/null 2>&1; then | ||
| # error_exit "Docker daemon is not running. Please start Docker and try again. | ||
| # You can start Docker with: sudo systemctl start docker" | ||
| # fi | ||
|
|
||
| # info "Docker daemon is running." | ||
|
|
||
| # # Check if the user can run Docker without sudo | ||
| # if docker ps > /dev/null 2>&1; then | ||
| # DOCKER_CMD="docker" | ||
| # else | ||
| # warning "Current user cannot run Docker commands without sudo." | ||
| # read -p "Do you want to run the Docker command with sudo? [y/N]: " USE_SUDO | ||
| # case "$USE_SUDO" in | ||
| # [yY][eE][sS]|[yY]) DOCKER_CMD="sudo docker" ;; | ||
| # *) error_exit "Docker command requires sudo. Exiting." ;; | ||
| # esac | ||
| # fi | ||
|
|
||
| # # Check if the Docker image exists locally | ||
| # IMAGE_NAME="yolov8_detection:latest" | ||
| # if ! $DOCKER_CMD image inspect "$IMAGE_NAME" > /dev/null 2>&1; then | ||
| # warning "Docker image '$IMAGE_NAME' not found locally." | ||
| # read -p "Do you want to build the Docker image now? [Y/n]: " BUILD_IMG | ||
| # BUILD_IMG=${BUILD_IMG:-Y} # Default to 'Y' if empty | ||
| # case "$BUILD_IMG" in | ||
| # [yY][eE][sS]|[yY]|"") | ||
| # info "Building Docker image '$IMAGE_NAME'..." | ||
| # $DOCKER_CMD build -t "$IMAGE_NAME" . | ||
| # info "Docker image '$IMAGE_NAME' built successfully." | ||
| # ;; | ||
| # *) | ||
| # error_exit "Docker image '$IMAGE_NAME' is required. Exiting." | ||
| # ;; | ||
| # esac | ||
| # else | ||
| # info "Docker image '$IMAGE_NAME' is available locally." | ||
| # fi | ||
|
|
||
| # # Check if the script is running on Linux (since --network=host is Linux-only) | ||
| # OS_NAME=$(uname) | ||
| # if [ "$OS_NAME" != "Linux" ]; then | ||
| # warning "--network=host is only supported on Linux. Adjusting network settings accordingly." | ||
| # # Optionally, you can set a different network or notify the user to modify the script | ||
| # # For simplicity, we'll proceed without --network=host | ||
| # NETWORK_OPTION="" | ||
| # else | ||
| # NETWORK_OPTION="--network=host" | ||
| # fi | ||
|
|
||
| # # Check if GPUs are available (optional, requires nvidia-docker) | ||
| # if command -v nvidia-smi &> /dev/null; then | ||
| # info "NVIDIA GPUs detected." | ||
| # GPU_OPTION="--gpus all" | ||
| # else | ||
| # warning "No NVIDIA GPUs detected or NVIDIA drivers not installed." | ||
| # GPU_OPTION="" | ||
| # fi | ||
|
|
||
| # # Optional: Check if Nginx RTMP server is accessible | ||
| # # Uncomment the following block if you want to perform this check | ||
| # : <<'END' | ||
| # NGINX_URL="rtmp://127.0.0.1:1935/live/stream" | ||
|
|
||
| # # Function to check if the RTMP server is reachable | ||
| # check_nginx_rtmp() { | ||
| # # Attempt to establish a TCP connection to the RTMP server | ||
| # nc -zv 127.0.0.1 1935 &> /dev/null | ||
| # if [ $? -ne 0 ]; then | ||
| # warning "Nginx RTMP server is not accessible at $NGINX_URL." | ||
| # read -p "Do you want to proceed anyway? [y/N]: " PROCEED | ||
| # case "$PROCEED" in | ||
| # [yY][eE][sS]|[yY]) ;; | ||
| # *) error_exit "Nginx RTMP server is required. Exiting." ;; | ||
| # esac | ||
| # else | ||
| # info "Nginx RTMP server is accessible at $NGINX_URL." | ||
| # fi | ||
| # } | ||
|
|
||
| # check_nginx_rtmp | ||
| # END | ||
|
|
||
| # # Run the Docker container | ||
| # info "Starting Docker container '$IMAGE_NAME'..." | ||
| # $DOCKER_CMD run $GPU_OPTION $NETWORK_OPTION | ||
|
|
||
| # # run with: | ||
| # # $DOCKER_CMD run $GPU_OPTION $NETWORK_OPTION --rm "$IMAGE_NAME" | ||
| # # if you want to delete after running | ||
|
|
||
| # info "Docker container '$IMAGE_NAME' has stopped." |