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

feat(docker): add option to run ollama w/o GPU support

Browse source
This commit is contained in:
Tanvir 2024-01-09 13:27:39 +06:00
parent e6e05d4641
commit 0c0c5955c6
No known key found for this signature in database
GPG key ID: 938B0F2643FE5107
1 changed files with 10 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

11
run-ollama-docker.sh
View file

@ -1,9 +1,14 @@
#!/bin/bash
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
# CPU Only
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
# GPU Support
# docker run -d --gpus=all -v ollama:/root/.ollama -p 11434:11434 --name ollama ollama/ollama
fi
docker image prune -f