David Krajewski commited on
Commit
17087b8
·
1 Parent(s): beb44b9

print nvidia folder dir

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -653,9 +653,11 @@ with gr.Blocks() as demo:
653
  # # print(os.path.dirname(nvidia.cublas.lib.__file__))
654
  # print(os.path.dirname(nvidia.cudnn.lib.__file__))
655
 
 
 
656
  import os
657
 
658
- folder_path = "/usr/local/cuda-11.7/lib64"
659
  if os.path.exists(folder_path) and os.path.isdir(folder_path):
660
  print(f"Contents of the folder '{folder_path}':")
661
  for item in os.listdir(folder_path):
 
653
  # # print(os.path.dirname(nvidia.cublas.lib.__file__))
654
  # print(os.path.dirname(nvidia.cudnn.lib.__file__))
655
 
656
+ # /usr/local/lib/python3.10/site-packages/nvidia/cudnn/lib
657
+
658
  import os
659
 
660
+ folder_path = "/usr/local/lib/python3.10/site-packages/nvidia/"
661
  if os.path.exists(folder_path) and os.path.isdir(folder_path):
662
  print(f"Contents of the folder '{folder_path}':")
663
  for item in os.listdir(folder_path):