diff --git a/run-ollama-docker.sh b/run-ollama-docker.sh
index 914975f4..20770a2d 100644
--- a/run-ollama-docker.sh
+++ b/run-ollama-docker.sh
@@ -1,9 +1,14 @@
 #!/bin/bash
 
+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
-# CPU Only
-docker run -d -v ollama:/root/.ollama -p 11434:11434 --name ollama ollama/ollama
-# GPU Support
-# docker run -d --gpus=all -v ollama:/root/.ollama -p 11434:11434 --name ollama ollama/ollama
-docker image prune -f
\ No newline at end of file
+
+if [ "$use_gpu" == "y" ]; then
+    docker run -d --gpus=all -v ollama:/root/.ollama -p 11434:11434 --name ollama ollama/ollama
+else
+    docker run -d -v ollama:/root/.ollama -p 11434:11434 --name ollama ollama/ollama
+fi
+
+docker image prune -f