Update app.py

app.py

@@ -13,21 +13,21 @@ def run_command(command):
         exit(1)
     return output.decode('utf-8')
 
-# Download CUDA installer
-download_command = "wget https://developer.download.nvidia.com/compute/cuda/12.2.0/local_installers/cuda_12.2.0_535.54.03_linux.run"
-result = run_command(download_command)
-if result is None:
-    print("Failed to download CUDA installer.")
-    exit(1)
-
-# Run CUDA installer without sudo
-install_command = "sh cuda_12.2.0_535.54.03_linux.run"
-result = run_command(install_command)
-if result is None:
-    print("Failed to run CUDA installer.")
-    exit(1)
-
-print("CUDA installation process completed.")
+commands = [
+    "wget https://developer.download.nvidia.com/compute/cuda/repos/debian11/x86_64/cuda-keyring_1.1-1_all.deb",
+    "dpkg -i cuda-keyring_1.1-1_all.deb",
+    "add-apt-repository contrib",
+    "apt-get update",
+    "apt-get -y install cuda"
+]
+
+for command in commands:
+    print(f"Executing: {command}")
+    if not run_command(command):
+        print("Installation failed.")
+        sys.exit(1)
+
+print("CUDA installation completed successfully.")
 
 def install_packages():