Printing CUDA_HOME location

app.py

@@ -23,6 +23,7 @@ import os
 import subprocess
 from subprocess import call
 import shlex
+import shutil
 os.environ["GRADIO_TEMP_DIR"] = os.path.join(os.getcwd(), "tmp")
 cwd = os.getcwd()
 print("Current working directory:", cwd)
@@ -51,6 +52,26 @@ with open('./build_ops.sh', 'rb') as file:
     script = file.read()
 call(script, shell=True)
 
+def find_cuda():
+    # Check if CUDA_HOME or CUDA_PATH environment variables are set
+    cuda_home = os.environ.get('CUDA_HOME') or os.environ.get('CUDA_PATH')
+
+    if cuda_home and os.path.exists(cuda_home):
+        return cuda_home
+
+    # Search for the nvcc executable in the system's PATH
+    nvcc_path = shutil.which('nvcc')
+
+    if nvcc_path:
+        # Remove the 'bin/nvcc' part to get the CUDA installation path
+        cuda_path = os.path.dirname(os.path.dirname(nvcc_path))
+        return cuda_path
+
+    return None
+
+cuda_path = find_cuda()
+print("Cuda path: " + str(cuda_path))
+
 class AppSteps(Enum):
     JUST_TEXT = 1
     TEXT_AND_EXEMPLARS = 2