mirror of
https://github.com/open-webui/open-webui
synced 2024-11-16 05:24:02 +00:00
feat(docker): add option to run ollama w/o GPU support
This commit is contained in:
parent
e6e05d4641
commit
0c0c5955c6
@ -1,9 +1,14 @@
|
|||||||
#!/bin/bash
|
#!/bin/bash
|
||||||
|
|
||||||
|
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
|
docker pull ollama/ollama
|
||||||
# CPU Only
|
|
||||||
docker run -d -v ollama:/root/.ollama -p 11434:11434 --name ollama ollama/ollama
|
if [ "$use_gpu" == "y" ]; then
|
||||||
# GPU Support
|
docker run -d --gpus=all -v ollama:/root/.ollama -p 11434:11434 --name ollama ollama/ollama
|
||||||
# docker run -d --gpus=all -v ollama:/root/.ollama -p 11434:11434 --name ollama ollama/ollama
|
else
|
||||||
docker image prune -f
|
docker run -d -v ollama:/root/.ollama -p 11434:11434 --name ollama ollama/ollama
|
||||||
|
fi
|
||||||
|
|
||||||
|
docker image prune -f
|
||||||
|
Loading…
Reference in New Issue
Block a user