diff --git a/docs/static/install.sh b/docs/static/install.sh index b631058c7e4e..8095580ea929 100755 --- a/docs/static/install.sh +++ b/docs/static/install.sh @@ -635,6 +635,13 @@ install_docker() { $SUDO systemctl start docker fi + # Avoid problem on older docker versions + DOCKER_VERSION=`docker version --format '{{.Server.Version}}' | sed 's/^\([0-9]*\.[0-9]*\).*/\1/'` + CONTAINER_PRIVILEGES= + if expr $DOCKER_VERSION '<' 20.10 >/dev/null ; then + CONTAINER_PRIVILEGES="--privileged=true" + fi + info "Creating LocalAI Docker volume..." # Create volume if doesn't exist already if ! $SUDO docker volume inspect local-ai-data > /dev/null 2>&1; then @@ -663,6 +670,7 @@ install_docker() { $SUDO docker run -v local-ai-data:/models \ --device /dev/dri \ --restart=always \ + ${CONTAINER_PRIVILEGES} \ -e API_KEY=$API_KEY \ -e THREADS=$THREADS \ $envs \ @@ -677,8 +685,8 @@ install_docker() { info "Checking Nvidia Kernel Drivers presence..." if ! available nvidia-smi; then - OS_NAME=$ID - OS_VERSION=$VERSION_ID + OS_NAME=$ID + OS_VERSION=$VERSION_ID case $OS_NAME in debian|ubuntu) $SUDO apt-get -y install nvidia-cuda-toolkit;; @@ -690,6 +698,7 @@ install_docker() { $SUDO docker run -v local-ai-data:/models \ --gpus all \ --restart=always \ + ${CONTAINER_PRIVILEGES} \ -e API_KEY=$API_KEY \ -e THREADS=$THREADS \ $envs \ @@ -707,6 +716,7 @@ install_docker() { --device /dev/kfd \ --group-add=video \ --restart=always \ + ${CONTAINER_PRIVILEGES} \ -e API_KEY=$API_KEY \ -e THREADS=$THREADS \ $envs \ @@ -722,6 +732,7 @@ install_docker() { $SUDO docker run -v local-ai-data:/models \ --device /dev/dri \ --restart=always \ + ${CONTAINER_PRIVILEGES} \ -e API_KEY=$API_KEY \ -e THREADS=$THREADS \ $envs \ @@ -737,12 +748,13 @@ install_docker() { info "Starting LocalAI Docker container..." $SUDO docker run -v local-ai-data:/models \ - --restart=always \ - -e MODELS_PATH=/models \ - -e API_KEY=$API_KEY \ - -e THREADS=$THREADS \ - $envs \ - -d -p $PORT:8080 --name local-ai localai/localai:$IMAGE_TAG $STARTCOMMAND + --restart=always \ + ${CONTAINER_PRIVILEGES} \ + -e MODELS_PATH=/models \ + -e API_KEY=$API_KEY \ + -e THREADS=$THREADS \ + $envs \ + -d -p $PORT:8080 --name local-ai localai/localai:$IMAGE_TAG $STARTCOMMAND fi install_success