浏览代码

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

Tanvir 1 年之前
父节点
当前提交
db51c22093
共有 1 个文件被更改,包括 4 次插入1 次删除
  1. 4 1
      run-ollama-docker.sh

+ 4 - 1
run-ollama-docker.sh

@@ -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"