UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-11-16 05:24:02 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

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
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