print dirs

app.py

@@ -647,19 +647,11 @@ class Drag:
 
 
 with gr.Blocks() as demo:
-    print(f"Current CUDA version: {torch.version.cuda}")
-    # import subprocess
-
-    # def find_file(filename):
-    #     result = subprocess.run(['find', '/', '-name', filename], stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
-    #     if result.returncode == 0:
-    #         print(f"Found the file '{filename}' at the following locations:")
-    #         print(result.stdout)
-    #     else:
-    #         print(f"Error occurred while searching for the file '{filename}':")
-    #         print(result.stderr)
-
-    # find_file("libcusolver.so.11")
+    import os
+    import nvidia.cublas.lib
+    import nvidia.cudnn.lib
+    print(os.path.dirname(nvidia.cublas.lib.__file__) + ":" + os.path.dirname(nvidia.cudnn.lib.__file__))
+
 
     gr.Markdown("""<h1 align="center">MOFA-Video</h1><br>""")