From f87a1c95435832ad21f5e32ac57bf1bd5ae4c80c Mon Sep 17 00:00:00 2001 From: Tanvir Date: Tue, 9 Jan 2024 13:30:58 +0600 Subject: [PATCH] refactor: update run-ollama-docker.sh --- run-ollama-docker.sh | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/run-ollama-docker.sh b/run-ollama-docker.sh index fd2a8eb2f..89bf10be0 100644 --- a/run-ollama-docker.sh +++ b/run-ollama-docker.sh @@ -1,14 +1,16 @@ #!/bin/bash -read -r -p "Do you want ollama in docker with GPU support? (y/n): " use_gpu +read -r -p "Do you want ollama in Docker with GPU support? (y/n): " use_gpu docker rm -f ollama || true docker pull ollama/ollama:latest +docker_args="-d -v ollama:/root/.ollama -p 11434:11434 --name ollama ollama/ollama" + if [ "$use_gpu" == "y" ]; then - docker run -d --gpus=all -v ollama:/root/.ollama -p 11434:11434 --name ollama ollama/ollama -else - docker run -d -v ollama:/root/.ollama -p 11434:11434 --name ollama ollama/ollama + docker_args+=" --gpus=all" fi +docker run "$docker_args" + docker image prune -f