mirror of
https://github.com/mudler/LocalAI.git
synced 2025-05-20 18:45:00 +00:00
feat(install): enhance script with choice functions and logs
- Added custom `choice_info`, `choice_warn`, and `choice_fatal` functions for interactive input logging. - Adjusted Docker volume creation message for better clarity. - Included NVIDIA driver check log for improved feedback to users. - Added consistent logging before starting LocalAI Docker containers across configurations. Signed-off-by: Alessandro Pirastru <alessandro.pirastru.94@gmail.com>
This commit is contained in:
parent
a7f70b0f44
commit
4c6aceac72
1 changed files with 25 additions and 1 deletions
26
docs/static/install.sh
vendored
26
docs/static/install.sh
vendored
|
@ -54,6 +54,24 @@ fatal()
|
||||||
exit 1
|
exit 1
|
||||||
}
|
}
|
||||||
|
|
||||||
|
# --- custom choice functions ---
|
||||||
|
# like the logging functions, but with the -n flag to prevent the new line and keep the cursor in line for choices inputs like y/n
|
||||||
|
choice_info()
|
||||||
|
{
|
||||||
|
echo -e -n "${BOLD}${LIGHT_BLUE}" '[INFO] ' "$@" "${RESET}"
|
||||||
|
}
|
||||||
|
|
||||||
|
choice_warn()
|
||||||
|
{
|
||||||
|
echo -e -n "${BOLD}${ORANGE}" '[WARN] ' "$@" "${RESET}" >&2
|
||||||
|
}
|
||||||
|
|
||||||
|
choice_fatal()
|
||||||
|
{
|
||||||
|
echo -e -n "${BOLD}${RED}" '[ERROR] ' "$@" "${RESET}" >&2
|
||||||
|
exit 1
|
||||||
|
}
|
||||||
|
|
||||||
# --- fatal if no systemd or openrc ---
|
# --- fatal if no systemd or openrc ---
|
||||||
verify_system() {
|
verify_system() {
|
||||||
if [ -x /sbin/openrc-run ]; then
|
if [ -x /sbin/openrc-run ]; then
|
||||||
|
@ -492,7 +510,7 @@ install_docker() {
|
||||||
$SUDO systemctl start docker
|
$SUDO systemctl start docker
|
||||||
fi
|
fi
|
||||||
|
|
||||||
info "Starting LocalAI Docker container..."
|
info "Creating LocalAI Docker volume..."
|
||||||
# Create volume if doesn't exist already
|
# Create volume if doesn't exist already
|
||||||
if ! $SUDO docker volume inspect local-ai-data > /dev/null 2>&1; then
|
if ! $SUDO docker volume inspect local-ai-data > /dev/null 2>&1; then
|
||||||
$SUDO docker volume create local-ai-data
|
$SUDO docker volume create local-ai-data
|
||||||
|
@ -533,6 +551,7 @@ install_docker() {
|
||||||
IMAGE_TAG=${LOCALAI_VERSION}-aio-gpu-nvidia-cuda-12
|
IMAGE_TAG=${LOCALAI_VERSION}-aio-gpu-nvidia-cuda-12
|
||||||
fi
|
fi
|
||||||
|
|
||||||
|
info "Checking Nvidia Kernel Drivers presence..."
|
||||||
if ! available nvidia-smi; then
|
if ! available nvidia-smi; then
|
||||||
#TODO Temporary Bypass for Fedora Headless (Cloud Edition), need to find a way to install nvidia-smi without pulling x11
|
#TODO Temporary Bypass for Fedora Headless (Cloud Edition), need to find a way to install nvidia-smi without pulling x11
|
||||||
OS_NAME=$ID
|
OS_NAME=$ID
|
||||||
|
@ -543,6 +562,7 @@ install_docker() {
|
||||||
esac
|
esac
|
||||||
fi
|
fi
|
||||||
|
|
||||||
|
info "Starting LocalAI Docker container..."
|
||||||
$SUDO docker run -v local-ai-data:/build/models \
|
$SUDO docker run -v local-ai-data:/build/models \
|
||||||
--gpus all \
|
--gpus all \
|
||||||
--restart=always \
|
--restart=always \
|
||||||
|
@ -561,6 +581,7 @@ install_docker() {
|
||||||
IMAGE_TAG=${LOCALAI_VERSION}-aio-gpu-hipblas
|
IMAGE_TAG=${LOCALAI_VERSION}-aio-gpu-hipblas
|
||||||
fi
|
fi
|
||||||
|
|
||||||
|
info "Starting LocalAI Docker container..."
|
||||||
$SUDO docker run -v local-ai-data:/build/models \
|
$SUDO docker run -v local-ai-data:/build/models \
|
||||||
--device /dev/dri \
|
--device /dev/dri \
|
||||||
--device /dev/kfd \
|
--device /dev/kfd \
|
||||||
|
@ -580,6 +601,7 @@ install_docker() {
|
||||||
IMAGE_TAG=${LOCALAI_VERSION}-aio-gpu-intel-f32
|
IMAGE_TAG=${LOCALAI_VERSION}-aio-gpu-intel-f32
|
||||||
fi
|
fi
|
||||||
|
|
||||||
|
info "Starting LocalAI Docker container..."
|
||||||
$SUDO docker run -v local-ai-data:/build/models \
|
$SUDO docker run -v local-ai-data:/build/models \
|
||||||
--device /dev/dri \
|
--device /dev/dri \
|
||||||
--restart=always \
|
--restart=always \
|
||||||
|
@ -597,6 +619,8 @@ install_docker() {
|
||||||
if [ "$USE_AIO" = true ]; then
|
if [ "$USE_AIO" = true ]; then
|
||||||
IMAGE_TAG=${LOCALAI_VERSION}-aio-cpu
|
IMAGE_TAG=${LOCALAI_VERSION}-aio-cpu
|
||||||
fi
|
fi
|
||||||
|
|
||||||
|
info "Starting LocalAI Docker container..."
|
||||||
$SUDO docker run -v local-ai-data:/models \
|
$SUDO docker run -v local-ai-data:/models \
|
||||||
--restart=always \
|
--restart=always \
|
||||||
-e MODELS_PATH=/models \
|
-e MODELS_PATH=/models \
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue