yueyang / run-ollama-docker.sh
github-actions[bot]
GitHub deploy: 45fae65ec18d0b265b7151dd469053c8494a72e8
ccaf67e
raw
history blame contribute delete
412 Bytes
#!/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 $host_port:$container_port --name ollama ollama/ollama"
if [ "$use_gpu" = "y" ]; then
docker_args="--gpus=all $docker_args"
fi
docker run $docker_args
docker image prune -f