talalif commited on
Commit
2bfdeb5
·
verified ·
1 Parent(s): 6fe31b9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -7,10 +7,15 @@ from PIL import Image
7
  import numpy as np
8
  from RealESRGAN import RealESRGAN
9
 
10
- os.environ['CUDA_HOME'] = "usr/local/cuda"
 
11
  os.environ['PATH'] = '/usr/local/lib/python3.10/site-packages/nvidia/cuda_nvrtc/lib:' + os.environ.get('PATH', '')
12
  os.environ['LD_LIBRARY_PATH'] = '/usr/local/lib/python3.10/site-packages/nvidia/cuda_nvrtc/lib:' + os.environ.get('LD_LIBRARY_PATH', '')
13
 
 
 
 
 
14
  print(os.environ['PATH'])
15
  print(os.environ['LD_LIBRARY_PATH'])
16
 
 
7
  import numpy as np
8
  from RealESRGAN import RealESRGAN
9
 
10
+ os.environ['CUDA_HOME'] = "/usr/local/lib/python3.10/site-packages/nvidia/cuda_runtime/include"
11
+
12
  os.environ['PATH'] = '/usr/local/lib/python3.10/site-packages/nvidia/cuda_nvrtc/lib:' + os.environ.get('PATH', '')
13
  os.environ['LD_LIBRARY_PATH'] = '/usr/local/lib/python3.10/site-packages/nvidia/cuda_nvrtc/lib:' + os.environ.get('LD_LIBRARY_PATH', '')
14
 
15
+ os.environ['PATH'] = '/usr/local/lib/python3.10/site-packages/nvidia/cuda_runtime/include:' + os.environ.get('PATH', '')
16
+ os.environ['LD_LIBRARY_PATH'] = '/usr/local/lib/python3.10/site-packages/nvidia/cuda_runtime/include:' + os.environ.get('LD_LIBRARY_PATH', '')
17
+
18
+
19
  print(os.environ['PATH'])
20
  print(os.environ['LD_LIBRARY_PATH'])
21