fix: run-ollama-docker.sh
This commit is contained in:
parent
ee675e794f
commit
685d37418f
|
@ -10,10 +10,10 @@ docker pull ollama/ollama:latest
|
||||||
|
|
||||||
docker_args="-d -v ollama:/root/.ollama -p $host_port:$container_port --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 $docker_args"
|
||||||
fi
|
fi
|
||||||
|
|
||||||
docker run "$docker_args"
|
docker run $docker_args
|
||||||
|
|
||||||
docker image prune -f
|
docker image prune -f
|
Loading…
Reference in New Issue