forked from open-webui/open-webui
refactor(run-ollama-docker.sh): update with port variables
This commit is contained in:
parent
fd33f68f28
commit
db51c22093
1 changed files with 4 additions and 1 deletions
|
@ -1,11 +1,14 @@
|
||||||
#!/bin/bash
|
#!/bin/bash
|
||||||
|
|
||||||
|
host_port=11434
|
||||||
|
container_port=11434
|
||||||
|
|
||||||
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"
|
docker_args="-d -v ollama:/root/.ollama -p $host_port:$container_port --name ollama ollama/ollama"
|
||||||
|
|
||||||
if [ "$use_gpu" == "y" ]; then
|
if [ "$use_gpu" == "y" ]; then
|
||||||
docker_args+=" --gpus=all"
|
docker_args+=" --gpus=all"
|
||||||
|
|
Loading…
Reference in a new issue