malleswar commited on
Commit
5f2cc26
·
verified ·
1 Parent(s): f0d6d6c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -2,6 +2,13 @@ current_path = os.environ.get('LD_LIBRARY_PATH', '')
2
  new_paths = '/usr/local/nvidia/lib:/usr/local/nvidia/lib64:/lib/x86_64-linux-gnu/'
3
 
4
 
 
 
 
 
 
 
 
5
  import gradio as gr
6
  import torch
7
  import subprocess
 
2
  new_paths = '/usr/local/nvidia/lib:/usr/local/nvidia/lib64:/lib/x86_64-linux-gnu/'
3
 
4
 
5
+ if current_path:
6
+ os.environ['LD_LIBRARY_PATH'] = f"{new_paths}:{current_path}"
7
+ else:
8
+ os.environ['LD_LIBRARY_PATH'] = new_paths
9
+
10
+ print(f"LD_LIBRARY_PATH: {os.environ['LD_LIBRARY_PATH']}")
11
+
12
  import gradio as gr
13
  import torch
14
  import subprocess