tdpeuter/open-webui
tdpeuter/open-webui
1
Fork 0
forked from open-webui/open-webui
Code Issues Pull requests 1 Releases 2 Packages Activity Actions

refactor: update run-ollama-docker.sh

Browse source
This commit is contained in:
Tanvir 2024-01-09 13:30:58 +06:00
parent 93c73d355b
commit f87a1c9543
No known key found for this signature in database
GPG key ID: 938B0F2643FE5107
1 changed files with 6 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
run-ollama-docker.sh
View file

@ -1,14 +1,16 @@
#!/bin/bash #!/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 rm -f ollama || true
docker pull ollama/ollama:latest docker pull ollama/ollama:latest
docker_args="-d -v ollama:/root/.ollama -p 11434:11434 --name ollama ollama/ollama"
if [ "$use_gpu" == "y" ]; then if [ "$use_gpu" == "y" ]; then
docker run -d --gpus=all -v ollama:/root/.ollama -p 11434:11434 --name ollama ollama/ollama docker_args+=" --gpus=all"
else
docker run -d -v ollama:/root/.ollama -p 11434:11434 --name ollama ollama/ollama
fi fi
docker run "$docker_args"
docker image prune -f docker image prune -f