diff --git a/install.sh b/install.sh
index 7b5fca4..1fdd68b 100644
--- a/install.sh
+++ b/install.sh
@@ -9,7 +9,7 @@ echo "Installing ffmpeg and cmake..."
 conda install ffmpeg cmake
 
 echo "Checking for CUDA installation..."
-if command -v nvcc &> /dev/null; then
+if command -v nvidia-smi &> /dev/null; then
     USE_CUDA=true
     echo "CUDA found."
 else