Update app.py

app.py

@@ -13,18 +13,21 @@ def run_command(command):
         exit(1)
     return output.decode('utf-8')
 
-commands = [
-    "wget https://developer.download.nvidia.com/compute/cuda/12.2.0/local_installers/cuda_12.2.0_535.54.03_linux.run",
-    "sh cuda_12.2.0_535.54.03_linux.run
-]
-
-for command in commands:
-    print(f"Executing: {command}")
-    if not run_command(command):
-        print("Installation failed.")
-        sys.exit(1)
-
-print("CUDA installation completed successfully.")
+# 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 in silent mode
+install_command = "sh cuda_12.2.0_535.54.03_linux.run --silent --toolkit --samples --override"
+result = run_command(install_command)
+if result is None:
+    print("Failed to run CUDA installer.")
+    exit(1)
+
+print("CUDA installation process completed.")
 
 def install_packages():