diff --git a/run.sh b/run.sh
index bd5b32e..7448232 100755
--- a/run.sh
+++ b/run.sh
@@ -1,12 +1,19 @@
 #!/bin/bash
 
-# Check if docker compose is installed
+# Check if docker is installed
 if ! command -v docker &> /dev/null
 then
     echo "Docker could not be found. Please install Docker and try again."
     exit
 fi
 
+# Check if docker composeis installed
+if ! docker compose version &> /dev/null
+then
+    echo "\"Docker compose\" could not be found. Please install \"Docker compose\" and try again."
+    exit
+fi
+
 # Parse command line arguments for model value and check for --with-cuda flag
 with_cuda=0
 while [[ "$#" -gt 0 ]]; do