refactor(run-ollama-docker.sh): update with port variables

This commit is contained in:
Tanvir 2024-01-09 13:46:07 +06:00
parent fd33f68f28
commit db51c22093
No known key found for this signature in database
GPG key ID: 938B0F2643FE5107

View file

@ -1,11 +1,14 @@
#!/bin/bash
host_port=11434
container_port=11434
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"
docker_args="-d -v ollama:/root/.ollama -p $host_port:$container_port --name ollama ollama/ollama"
if [ "$use_gpu" == "y" ]; then
docker_args+=" --gpus=all"