talalif commited on
Commit
e1a6a97
·
verified ·
1 Parent(s): 6311fef

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ from PIL import Image
7
  import numpy as np
8
  from RealESRGAN import RealESRGAN
9
 
10
- os.environ['CUDA_HOME'] = "/usr/local/lib/python3.10/site-packages/cupy/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
 
 
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