Procházet zdrojové kódy

check cuda installed before installing

Michael Yang před 1 rokem
rodič
revize
5e388f931e
1 změnil soubory, kde provedl 8 přidání a 0 odebrání
  1. 8 0
      scripts/install.sh

+ 8 - 0
scripts/install.sh

@@ -117,6 +117,11 @@ check_gpu() {
     esac
 }
 
+if check_gpu nvidia-smi; then
+    status "NVIDIA GPU installed."
+    exit 0
+fi
+
 if ! check_gpu lspci && ! check_gpu lshw; then
     warning "No NVIDIA GPU detected. Ollama will run in CPU-only mode."
     exit 0
@@ -228,3 +233,6 @@ if ! lsmod | grep -q nvidia; then
 
     $SUDO modprobe nvidia
 fi
+
+
+status "NVIDIA GPU installed."